Update app.py

app.py

@@ -2,14 +2,26 @@ import streamlit as st
 import numpy as np
 import pandas as pd
 import plotly.express as px
+from statsforecast import StatsForecast
+from statsforecast.models import seasonal_naive, auto_arima
+
+st.title('Forecasting time series')
 
 uploaded_file = st.file_uploader("Choose a CSV file", accept_multiple_files=False)
 if uploaded_file is not None:
-     # Can be used wherever a "file-like" object is accepted:
      dataframe = pd.read_csv(uploaded_file)
      st.write(dataframe)
      series = st.text_input("Write the name of the variable you want to forecast")
-     dataframe['time'] = dataframe.index
-     if series is not None:    
-          fig = px.line(dataframe, x='time', y=series, title=f'Forecasting {series}')
+     date = st.text_input("Write the first date")
+     freq = st.text_input("Write the frequency")
+     if series is not None:
+          series_train = pd.DataFrame(
+                        {
+                            'ds': pd.date_range(start=date, periods=dataframe.shape[0], freq=freq),
+                            'y': dataframe[series]
+                        },
+                        index=pd.Index([0] * dataframe.shape[0], name='unique_id')
+          )
+              
+          fig = px.line(series_train, x='ds', y='y', title=f'{series}')
           st.plotly_chart(fig)