Update app.py

app.py

@@ -13,13 +13,17 @@ def plot_forecast(final_year, companies, noise, show_legend, point_style):
     plt_format = ({"cross": "X", "line": "-", "circle": "o--"})[point_style]
     fig = plt.figure()
     ax = fig.add_subplot(111)
-    for i, company in enumerate(companies):
-        series = np.arange(0, year_count, dtype=float)
-        series = series**2 * (i + 1)
-        series += np.random.rand(year_count) * noise
-        ax.plot(x, series, plt_format)
+    for i, company in enumerate(["Google", "Microsoft", "Gradio", "Apple", "Amazon", "Facebook"]):
+        if company in companies:
+            series = np.arange(0, year_count, dtype=float)
+            series = series**2 * (i + 1)
+            series += np.random.rand(year_count) * noise
+            ax.plot(x, series, plt_format, label=company)
     if show_legend:
-        plt.legend(companies)
+        plt.legend()
+    ax.set_xlabel("Year")
+    ax.set_ylabel("Revenue")
+    ax.set_title("Revenue Forecast for Selected Companies")
     return fig