Guardar mis cambios locales

app.py

@@ -23,7 +23,7 @@ def parse_date(date_str):
     except ValueError:
         return None, None, "Date format should be 'Month-Year', e.g., 'January-2024'."
 
-def forecast_sales(uploaded_file, start_date_str, end_date_str, forecast_period=30):
+def forecast_sales(uploaded_file, start_date_str, end_date_str):
     if uploaded_file is None:
         return "No file uploaded.", None, "Please upload a file."
 
@@ -49,21 +49,18 @@ def forecast_sales(uploaded_file, start_date_str, end_date_str, forecast_period=
         return None, error, "Failed to load ARIMA model."
 
     try:
-        forecast_index = pd.date_range(start=end_date, periods=forecast_period + 1, freq='D')[1:]
-        forecast = arima_model.get_forecast(steps=forecast_period)
+        forecast = arima_model.get_forecast(steps=60)
+        forecast_index = pd.date_range(start=end_date, periods=61, freq='D')[1:]
         forecast_df = pd.DataFrame({'Date': forecast_index, 'Sales Forecast': forecast.predicted_mean})
-    except Exception as e:
-        return None, f"Failed during forecasting: {str(e)}", "Forecasting failed."
-
-    try:
+        
         fig, ax = plt.subplots(figsize=(10, 6))
-        ax.plot(df_filtered['ds'], df_filtered['y'], label='Actual Sales', color='red')
-        ax.plot(forecast_df['Date'], forecast_df['Sales Forecast'], label='Sales Forecast', color='blue', linestyle='--')
+        ax.plot(df_filtered['ds'], df_filtered['y'], label='Actual Sales', color='blue')
+        ax.plot(forecast_df['Date'], forecast_df['Sales Forecast'], label='Sales Forecast', color='red', linestyle='--')
         ax.set_xlabel('Date')
         ax.set_ylabel('Sales')
         ax.set_title('Sales Forecasting with ARIMA')
         ax.legend()
-        return None, fig, "File loaded and processed successfully."
+        return fig, "File loaded and processed successfully."
     except Exception as e:
         return None, f"Failed to generate plot: {str(e)}", "Plotting failed."
 
@@ -80,7 +77,7 @@ def setup_interface():
         forecast_button.click(
             forecast_sales,
             inputs=[file_input, start_date_input, end_date_input],
-            outputs=[output_message, output_plot]
+            outputs=[output_plot, output_message]
         )
     return demo