Update app.py

app.py

@@ -2,152 +2,115 @@ import streamlit as st
 import pandas as pd
 from datetime import datetime
 import plotly.express as px
+import plotly.graph_objects as go
+from io import BytesIO
 
-# Set page config
-st.set_page_config(page_title="مدیریت داده‌های مزارع", layout="wide", direction="rtl")
+# Page configuration
+st.set_page_config(page_title="مدیریت داده‌های مزرعه", layout="wide")
 
-# Add custom CSS for RTL support
-st.markdown(
-    """
-    <style>
-        .stSelectbox, .stMultiSelect {
-            direction: rtl;
-        }
-        div[data-testid="stDataFrame"] {
-            direction: rtl;
-        }
-    </style>
-    """, unsafe_allow_html=True
-)
+# Load data
+@st.cache_data
+def load_data():
+    try:
+        df = pd.read_csv('پایگاه داده.csv')
+        return df
+    except Exception as e:
+        st.error(f"Error loading data: {str(e)}")
+        return pd.DataFrame()
 
 # Initialize session state
 if 'data' not in st.session_state:
-    # Create sample DataFrame or load from CSV
-    st.session_state.data = pd.DataFrame({
-        'مزرعه': [], 'کانال': [], 'اداره': [], 'سن': [], 
-        'واریته': [], 'مساحت': [], 'مساحت زیرمجموعه': [], 
-        'گروه': [], 'ایام هفته': []
-    })
+    st.session_state.data = load_data()
 
-def validate_input(value, field_type):
-    """Validate input values based on field type"""
-    if field_type == 'numeric':
-        try:
-            val = float(value)
-            if val < 0:
-                return False, "مقدار نمی‌تواند منفی باشد"
-            return True, val
-        except:
-            return False, "لطفا یک عدد معتبر وارد کنید"
-    return True, value
+# Main title
+st.title("سیستم مدیریت داده‌های مزرعه")
 
-def calculate_growth(current_height, previous_height):
-    """Calculate growth rate"""
-    if current_height and previous_height:
-        return current_height - previous_height
-    return 0
+# Create three tabs
+tab1, tab2, tab3 = st.tabs(["اطلاعات پایه", "اطلاعات نیتروژن", "اطلاعات ایستگاه‌ها"])
 
-# Sidebar filters
-st.sidebar.header("فیلترها")
-
-# Week and day selection
-days = ["شنبه", "یکشنبه", "دوشنبه", "سه‌شنبه", "چهارشنبه", "پنجشنبه"]
-weeks = list(range(1, 23))
-
-selected_day = st.sidebar.selectbox("روز هفته", days)
-selected_week = st.sidebar.selectbox("هفته", weeks)
-
-# Main content
-st.title("سیستم مدیریت داده‌های مزارع")
-
-# Data entry form
-with st.form("data_entry"):
-    col1, col2, col3 = st.columns(3)
-    
-    with col1:
-        station1 = st.number_input("ایستگاه 1", min_value=0.0)
-        station2 = st.number_input("ایستگاه 2", min_value=0.0)
-        station3 = st.number_input("ایستگاه 3", min_value=0.0)
-        
-    with col2:
-        station4 = st.number_input("ایستگاه 4", min_value=0.0)
-        station5 = st.number_input("ایستگاه 5", min_value=0.0)
-        current_height = st.number_input("ارتفاع هفته جاری", min_value=0.0)
+with tab1:
+    # Basic Information Form
+    st.subheader("ورود اطلاعات پایه")
+    with st.form("basic_info_form"):
+        col1, col2, col3 = st.columns(3)
         
-    with col3:
-        previous_height = st.number_input("ارتفاع هفته گذشته", min_value=0.0)
-        nitrogen = st.number_input("نیتروژن فعلی", min_value=0.0)
-        moisture = st.number_input("رطوبت غلاف فعلی", min_value=0.0)
+        with col1:
+            farm_id = st.text_input("نماینده")
+            channel = st.text_input("کانال")
+            department = st.text_input("اداره")
+            
+        with col2:
+            production = st.text_input("تولید")
+            area = st.number_input("مساحت داشت", 0.0)
+            sub_area = st.number_input("مساحت زیرمجموعه", 0.0)
+            
+        with col3:
+            variety = st.text_input("واریته")
+            age = st.text_input("سن")
+            date = st.date_input("تاریخ برداشت")
+            
+        submit_basic = st.form_submit_button("ثبت اطلاعات پایه")
+        if submit_basic:
+            st.success("اطلاعات پایه با موفقیت ثبت شد")
 
-    submitted = st.form_submit_button("ثبت اطلاعات")
-    
-    if submitted:
-        # Validate and calculate
-        growth_rate = calculate_growth(current_height, previous_height)
-        
-        # Add to DataFrame
-        new_data = {
-            'ایستگاه 1': station1,
-            'ایستگاه 2': station2,
-            'ایستگاه 3': station3,
-            'ایستگاه 4': station4,
-            'ایستگاه 5': station5,
-            'ارتفاع هفته جاری': current_height,
-            'ارتفاع هفته گذشته': previous_height,
-            'رشد': growth_rate,
-            'نیتروژن': nitrogen,
-            'رطوبت': moisture,
-            'هفته': selected_week,
-            'روز': selected_day
-        }
+with tab2:
+    # Nitrogen Information Form
+    st.subheader("ورود اطلاعات نیتروژن")
+    with st.form("nitrogen_info_form"):
+        col1, col2 = st.columns(2)
         
-        # Update session state
-        st.session_state.data = pd.concat([
-            st.session_state.data, 
-            pd.DataFrame([new_data])
-        ], ignore_index=True)
+        with col1:
+            nitrogen_current = st.number_input("نیتروژن فعلی", 0.0)
+            nitrogen_std = st.number_input("نیتروژن استاندارد قبلی", 0.0)
+            nitrogen_prev = st.number_input("نیتروژن قبلی", 0.0)
+            
+        with col2:
+            nitrogen_std_prev = st.number_input("نیتروژن استاندارد قبلی فعلی", 0.0)
+            moisture_current = st.number_input("رطوبت فعلی", 0.0)
+            moisture_std = st.number_input("رطوبت استاندارد قبلی", 0.0)
+            
+        submit_nitrogen = st.form_submit_button("ثبت اطلاعات نیتروژن")
+        if submit_nitrogen:
+            st.success("اطلاعات نیتروژن با موفقیت ثبت شد")
+
+with tab3:
+    # Station Information Form
+    st.subheader("ورود اطلاعات ایستگاه‌ها")
+    with st.form("station_info_form"):
+        col1, col2, col3 = st.columns(3)
         
-        st.success("اطلاعات با موفقیت ثبت شد")
+        with col1:
+            station1 = st.number_input("ایستگاه ۱", 0.0)
+            station2 = st.number_input("ایستگاه ۲", 0.0)
+            station3 = st.number_input("ایستگاه ۳", 0.0)
+            
+        with col2:
+            station4 = st.number_input("ایستگاه ۴", 0.0)
+            station5 = st.number_input("ایستگاه ۵", 0.0)
+            height_current = st.number_input("ارتفاع هفته جاری", 0.0)
+            
+        with col3:
+            height_prev = st.number_input("ارتفاع هفته گذشته", 0.0)
+            growth_current = st.number_input("رشد هفته جاری", 0.0)
+            growth_prev = st.number_input("رشد هفته گذشته", 0.0)
+            
+        submit_station = st.form_submit_button("ثبت اطلاعات ایستگاه‌ها")
+        if submit_station:
+            st.success("اطلاعات ایستگاه‌ها با موفقیت ثبت شد")
 
-# Display data
-if not st.session_state.data.empty:
-    st.subheader("داده‌های ثبت شده")
-    st.dataframe(st.session_state.data)
-    
-    # Download button
-    csv = st.session_state.data.to_csv(index=False)
-    st.download_button(
-        label="دانلود داده‌ها",
-        data=csv,
-        file_name="پایگاه داده.csv",
-        mime="text/csv"
-    )
-    
-    # Visualizations
-    if len(st.session_state.data) > 0:
-        st.subheader("نمودار رشد")
-        fig = px.line(
-            st.session_state.data, 
-            x='هفته', 
-            y='ارتفاع هفته جاری',
-            title='روند رشد بر اساس هفته'
-        )
-        st.plotly_chart(fig)
+# Display data table
+st.header("نمایش داده‌ها")
+st.dataframe(st.session_state.data)
 
-# Add data filtering
-st.sidebar.subheader("فیلتر داده‌ها")
-if not st.session_state.data.empty:
-    filtered_data = st.session_state.data.copy()
-    
-    # Add filter widgets
-    for column in ['اداره', 'واریته', 'گروه']:
-        if column in filtered_data.columns:
-            values = ['همه'] + list(filtered_data[column].unique())
-            selected = st.sidebar.selectbox(f"فیلتر بر اساس {column}", values)
-            
-            if selected != 'همه':
-                filtered_data = filtered_data[filtered_data[column] == selected]
-    
-    if not filtered_data.empty:
-        st.subheader("داده‌های فیلتر شده")
-        st.dataframe(filtered_data)
+# Export functionality
+st.header("خروجی داده‌ها")
+if st.button("دانلود داده‌ها"):
+    output = BytesIO()
+    st.session_state.data.to_excel(output, index=False)
+    output.seek(0)
+    st.download_button(
+        label="دانلود فایل اکسل",
+        data=output,
+        file_name=f"farm_data_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx",
+        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
+    )