Update app.py

app.py

@@ -3,15 +3,31 @@ import pandas as pd
 import matplotlib.pyplot as plt
 import seaborn as sns
 import plotly.graph_objects as go
+from sklearn.model_selection import train_test_split
 from sklearn.linear_model import LinearRegression
-import numpy as np
 
-# عنوان برنامه
+# تنظیمات صفحه
 st.set_page_config(page_title="اداره زراعت و کنترل محصول", layout="wide")
 
+# تابع پیش‌بینی
+def predict_growth(crop_data, weeks_to_predict=8):
+    X = crop_data[['هفته', 'رشد']].dropna().values
+    y = crop_data['رشد'].dropna().values
+    
+    model = LinearRegression()
+    model.fit(X, y)
+    
+    future_weeks = [[week, crop_data[crop_data['هفته'] == week]['رشد'].mean()] for week in range(1, weeks_to_predict + 1)]
+    future_weeks_df = pd.DataFrame(future_weeks, columns=['هفته', 'رشد'])
+    
+    predictions = model.predict(future_weeks_df)
+    
+    future_weeks_df['پیش‌بینی رشد'] = predictions
+    return future_weeks_df
+
 # نوار کناری
 st.sidebar.title("منو")
-menu = st.sidebar.selectbox("انتخاب کنید", ["نمایش چارت‌ها", "پیش بینی رشد مزارع نیشکر با استفاده از ماشین لرنینگ", "چت بات"])
+menu = st.sidebar.selectbox("انتخاب کنید", ["نمایش چارت‌ها", "پیش‌بینی رشد مزارع نیشکر با استفاده از ماشین لرنینگ", "چت بات"])
 
 # چت بات
 if menu == "چت بات":
@@ -21,27 +37,21 @@ if menu == "چت بات":
       window.botpressWebChat.init({
         hostUrl: "https://cdn.botpress.cloud/webchat/v2",
         messagingUrl: "https://mediafiles.botpress.cloud/29d24895-b0fa-4941-a3c4-8afa638f8b28",
-        botId: "29d24895-b0fa-4941-a3c4-8afa638f8b28",
+        botId: "your-bot-id",
       })
     </script>
     """, unsafe_allow_html=True)
 
-# بارگذاری داده‌ها یکبار برای همیشه
-if 'crop_data' not in st.session_state:
-    uploaded_crop_file = st.sidebar.file_uploader("فایل داده‌های رشد و ارتفاع را آپلود کنید", type="csv")
-    if uploaded_crop_file:
-        st.session_state.crop_data = pd.read_csv(uploaded_crop_file)
-
-if 'weather_data' not in st.session_state:
-    uploaded_weather_file = st.sidebar.file_uploader("فایل داده‌های هواشناسی را آپلود کنید", type="csv")
-    if uploaded_weather_file:
-        st.session_state.weather_data = pd.read_csv(uploaded_weather_file)
-
 # نمایش چارت‌ها
-if menu == "نمایش چارت‌ها":
+elif menu == "نمایش چارت‌ها":
     st.title("نمایش چارت‌ها و نمودارهای رشد و ارتفاع مزارع نیشکر")
 
-    if 'crop_data' in st.session_state:
+    if 'crop_data' not in st.session_state:
+        uploaded_file = st.file_uploader("فایل داده‌های رشد و ارتفاع را آپلود کنید", type="csv")
+        if uploaded_file:
+            st.session_state.crop_data = pd.read_csv(uploaded_file)
+            st.write("داده‌های رشد و ارتفاع بارگذاری شد.")
+    else:
         crop_data = st.session_state.crop_data
 
         farms = crop_data['مزرعه'].unique()
@@ -57,5 +67,45 @@ if menu == "نمایش چارت‌ها":
         ax[0].set_xlabel("هفته")
         ax[0].set_ylabel("رشد")
 
-        sns.lineplot(x="هفته", y="ارتفاع", data=farm_data, ax=ax[
+        sns.lineplot(x="هفته", y="ارتفاع", data=farm_data, ax=ax[1])
+        ax[1].set_title("ارتفاع گیاه")
+        ax[1].set_xlabel("هفته")
+        ax[1].set_ylabel("ارتفاع")
+
+        st.pyplot(fig)
+
+        st.subheader("نمودار سه بعدی ارتفاع گیاه")
+        fig = go.Figure(data=[go.Surface(z=farm_data.pivot(index='هفته', columns='مزرعه', values='ارتفاع').values)])
+        fig.update_layout(title='نمودار سه بعدی ارتفاع گیاه', autosize=True,
+                          scene=dict(zaxis_title='ارتفاع', xaxis_title='هفته', yaxis_title='مزرعه'))
+        st.plotly_chart(fig)
+
+# پیش‌بینی رشد مزارع نیشکر با استفاده از ماشین لرنینگ
+elif menu == "پیش‌بینی رشد مزارع نیشکر با استفاده از ماشین لرنینگ":
+    st.title("پیش‌بینی رشد مزارع نیشکر با استفاده از ماشین لرنینگ")
+
+    if 'crop_data' not in st.session_state:
+        uploaded_file = st.file_uploader("فایل داده‌های رشد و ارتفاع را آپلود کنید", type="csv")
+        if uploaded_file:
+            st.session_state.crop_data = pd.read_csv(uploaded_file)
+            st.write("داده‌های رشد و ارتفاع بارگذاری شد.")
+    else:
+        crop_data = st.session_state.crop_data
+
+        farms = crop_data['مزرعه'].unique()
+        selected_farm = st.selectbox("انتخاب مزرعه", farms)
+        
+        farm_data = crop_data[crop_data['مزرعه'] == selected_farm]
+        predicted_growth = predict_growth(farm_data)
+
+        st.subheader("پیش‌بینی رشد برای هفته هشتم")
+        st.write(predicted_growth)
+
+        st.subheader("نمودار رشد پیش‌بینی شده")
+        fig, ax = plt.subplots(figsize=(10, 6))
+        sns.lineplot(x=predicted_growth['هفته'], y=predicted_growth['پیش‌بینی رشد'], ax=ax)
+        ax.set_title("رشد پیش‌بینی شده")
+        ax.set_xlabel("هفته")
+        ax.set_ylabel("رشد پیش‌بینی شده")
+        st.pyplot(fig)