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, date_str, forecast_period=30):
+def forecast_sales(uploaded_file, start_date_str, end_date_str, forecast_period=30):
     if uploaded_file is None:
         return "No file uploaded.", None, "Please upload a file."
 
@@ -34,13 +34,14 @@ def forecast_sales(uploaded_file, date_str, forecast_period=30):
     except Exception as e:
         return None, f"Failed to read the uploaded CSV file: {str(e)}", "Error reading file."
 
+    start_date, _, error = parse_date(start_date_str)
+    _, end_date, error_end = parse_date(end_date_str)
+    if error or error_end:
+        return None, error or error_end, "Invalid date format."
+
     df['Date'] = pd.to_datetime(df['Date'])
     df = df.rename(columns={'Date': 'ds', 'Sale': 'y'})
 
-    start_date, end_date, error = parse_date(date_str)
-    if error:
-        return None, error, "Invalid date format."
-
     df_filtered = df[(df['ds'] >= start_date) & (df['ds'] <= end_date)]
 
     arima_model, error = load_model()
@@ -48,8 +49,8 @@ def forecast_sales(uploaded_file, date_str, forecast_period=30):
         return None, error, "Failed to load ARIMA model."
 
     try:
-        forecast = arima_model.get_forecast(steps=forecast_period)
         forecast_index = pd.date_range(start=end_date, periods=forecast_period + 1, freq='D')[1:]
+        forecast = arima_model.get_forecast(steps=forecast_period)
         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."
@@ -62,7 +63,7 @@ def forecast_sales(uploaded_file, date_str, forecast_period=30):
         ax.set_ylabel('Sales')
         ax.set_title('Sales Forecasting with ARIMA')
         ax.legend()
-        return fig, None, "File loaded and processed successfully."
+        return None, fig, "File loaded and processed successfully."
     except Exception as e:
         return None, f"Failed to generate plot: {str(e)}", "Plotting failed."
 
@@ -71,14 +72,15 @@ def setup_interface():
         gr.Markdown("## MLCast v1.1 - Intelligent Sales Forecasting System")
         with gr.Row():
             file_input = gr.File(label="Upload your store data here (must contain Date and Sales)")
-            date_input = gr.Textbox(label="Enter Month and Year (e.g., January-2024)")
+            start_date_input = gr.Textbox(label="Start Date (e.g., January-2024)", placeholder="Enter Start Date")
+            end_date_input = gr.Textbox(label="End Date (e.g., January-2024)", placeholder="Enter End Date")
             forecast_button = gr.Button("Forecast Sales")
         output_plot = gr.Plot()
         output_message = gr.Textbox(label="System Messages", visible=True)
         forecast_button.click(
             forecast_sales,
-            inputs=[file_input, date_input],
-            outputs=[output_plot, output_message]
+            inputs=[file_input, start_date_input, end_date_input],
+            outputs=[output_message, output_plot]
         )
     return demo