Update app.py

app.py

@@ -7,7 +7,6 @@ import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.dates as mdates
 from sklearn.metrics import mean_absolute_error, mean_squared_error
-
 def get_popular_tickers():
     return [
         "AAPL", "MSFT", "GOOGL", "AMZN", "META", "TSLA", "NVDA", "JPM",
@@ -16,10 +15,6 @@ def get_popular_tickers():
 
 def predict_stock(ticker, train_data_points, prediction_days):
     try:
-        # Asegurar que los parámetros sean enteros
-        train_data_points = int(train_data_points)
-        prediction_days = int(prediction_days)
-        
         # Configurar el pipeline
         pipeline = ChronosPipeline.from_pretrained(
             "amazon/chronos-t5-mini",
@@ -89,6 +84,7 @@ def predict_stock(ticker, train_data_points, prediction_days):
         
         # Calcular métricas si hay datos reales para comparar
         overlap_end_index = train_data_points + prediction_days
+        validation_data = None
         if overlap_end_index <= total_points:
             real_future_dates = df['Date'][train_data_points:overlap_end_index]
             real_future_data = df[f'{ticker}_Close'][train_data_points:overlap_end_index].values
@@ -149,8 +145,10 @@ def predict_stock(ticker, train_data_points, prediction_days):
             prediction_df = prediction_df[prediction_df['Date'].isin(matching_dates)]
             prediction_df['Real_Price'] = real_future_data[:len(prediction_df)]
         
-        # Retornar el gráfico y los datos de predicción
-        return plt, gr.File.update(value=prediction_df.to_csv(index=False), filename=f"predictions_{ticker}.csv")
+        csv_path = f"predictions_{ticker}.csv"
+        prediction_df.to_csv(csv_path, index=False)
+        
+        return plt, csv_path
             
     except Exception as e:
         print(f"Error: {str(e)}")
@@ -168,7 +166,7 @@ with gr.Blocks() as demo:
             )
             train_data_points = gr.Slider(
                 minimum=50,
-                maximum=5000,
+                maximum=5000,  # Puedes ajustar este valor si lo deseas
                 value=1000,
                 step=1,
                 label="Número de Datos para Entrenamiento"
@@ -181,7 +179,7 @@ with gr.Blocks() as demo:
                 label="Número de Días a Predecir"
             )
             predict_btn = gr.Button("Predecir")
-        
+    
     with gr.Column():
         plot_output = gr.Plot(label="Gráfico de Predicción")
         download_btn = gr.File(label="Descargar Predicciones")
@@ -192,7 +190,7 @@ with gr.Blocks() as demo:
         hist = stock.history(period="max")
         total_points = len(hist)
         # Actualizar el deslizador para reflejar el número total de puntos disponibles
-        return gr.Slider.update(maximum=total_points, value=min(1000, total_points))
+        return gr.update(maximum=total_points, value=min(1000, total_points))
     
     ticker.change(
         fn=update_train_data_points,
@@ -206,4 +204,4 @@ with gr.Blocks() as demo:
         outputs=[plot_output, download_btn]
     )
 
-demo.launch()
+demo.launch(debug=True)