Update app.py

app.py

@@ -123,11 +123,12 @@ for company in companies:
 # --- FUNZIONE PER GRADIO ---
 def show_company_data(selected_companies, aggregation="Day"):
     if not selected_companies:
-        return "Seleziona almeno un'azienda", None
+        return "Seleziona almeno un'azienda", None, None
 
-    fig = go.Figure()
     agg_col = {"Day": "Day", "Month": "Month", "Year": "Year"}.get(aggregation, "Day")
 
+    # --- Grafico Strategie ---
+    fig_strat = go.Figure()
     for c in selected_companies:
         if c not in dfs_final:
             continue
@@ -136,47 +137,73 @@ def show_company_data(selected_companies, aggregation="Day"):
         df_grouped = df_c.groupby(agg_col).agg({
             'StrategyA_Cumulative':'last',
             'StrategyB_Cumulative':'last',
-            'StrategyC_Cumulative':'last',
-            f'Close_{TICKERS[c]}':'last'
+            'StrategyC_Cumulative':'last'
         }).reset_index()
 
-        # Strategie
         for strat in ['StrategyA_Cumulative','StrategyB_Cumulative','StrategyC_Cumulative']:
-            fig.add_trace(go.Scatter(
+            fig_strat.add_trace(go.Scatter(
                 x=df_grouped[agg_col], y=df_grouped[strat],
                 mode="lines", name=f"{c} {strat}"
             ))
 
-        # Prezzo (asse Y2)
-        fig.add_trace(go.Scatter(
+    fig_strat.update_layout(
+        title="Confronto Strategie (Sentiment, Regression, FinBERT)",
+        xaxis_title=aggregation,
+        yaxis_title="Cumulative",
+        template="plotly_dark",
+        hovermode="x unified"
+    )
+
+    # --- Grafico Prezzi ---
+    fig_price = go.Figure()
+    for c in selected_companies:
+        if c not in dfs_final:
+            continue
+        df_c = dfs_final[c]
+
+        df_grouped = df_c.groupby(agg_col).agg({
+            f'Close_{TICKERS[c]}':'last'
+        }).reset_index()
+
+        fig_price.add_trace(go.Scatter(
             x=df_grouped[agg_col], y=df_grouped[f'Close_{TICKERS[c]}'],
-            mode="lines", name=f"{c} Price", yaxis="y2", line=dict(dash="dot")
+            mode="lines", name=f"{c} Price"
         ))
 
-    fig.update_layout(
-        title="Confronto Strategie (Sentiment, Regression, FinBERT) + Prezzi",
+    fig_price.update_layout(
+        title="Andamento Prezzi",
         xaxis_title=aggregation,
-        yaxis_title="Cumulative",
-        yaxis2=dict(title="Stock Price", overlaying="y", side="right"),
-        legend=dict(orientation="h", yanchor="bottom", y=1.02, xanchor="right", x=1),
+        yaxis_title="Stock Price",
         template="plotly_dark",
         hovermode="x unified"
     )
-    return None, fig
+
+    return None, fig_strat, fig_price
 
 # --- INTERFACCIA GRADIO ---
 demo = gr.Interface(
     fn=show_company_data,
     inputs=[
-        gr.Dropdown(choices=companies, value=["Microsoft","Tesla"], multiselect=True, label="Seleziona Aziende"),
-        gr.Radio(choices=["Day","Month","Year"], value="Day", label="Aggregazione")
+        gr.Dropdown(
+            choices=companies,
+            value=["Microsoft", "Tesla"],
+            multiselect=True,
+            label="Select Companies"
+        ),
+        gr.Radio(
+            choices=["Day", "Month", "Year"],
+            value="Day",
+            label="Aggregation Level"
+        )
     ],
     outputs=[
-        gr.Dataframe(label="Preview", type="pandas"),
-        gr.Plot(label="Strategie & Prezzi")
+        gr.Dataframe(label="Data Preview", type="pandas"),
+        gr.Plot(label="Strategies"),
+        gr.Plot(label="Prices")
     ],
     title="Portfolio Evolution",
-    description="Confronto tra Sentiment Custom, Regressione e FinBERT con andamento prezzi"
+    description="Compare Custom Sentiment, Regression, and FinBERT strategies alongside stock prices."
 )
 
 demo.launch()
+