Update app.py

app.py

@@ -342,9 +342,11 @@ lottie_report = load_lottie_url('https://assets9.lottiefiles.com/packages/lf20_v
 # Create session state for storing data with Persian column names
 if 'heights_df' not in st.session_state:
     st.session_state.heights_df = pd.DataFrame(columns=[
-        'ردیف', 'هفته', 'تاریخ قرائت', 'ارتفاع هفته جاری مزرعه', 'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3',
-        'ایستگاه 4', 'ایستگاه 5', 'چاهک 1', 'چاهک 2', 'رطوبت غلاف فعلی', 'نیتروژن فعلی',
-        'واریته', 'سن', 'مساحت داشت', 'کانال', 'اداره'
+        'ردیف', 'نماینده', 'کانال', 'اداره', 'تولید', 'مساحت داشت', 'مساحت زیرمجموعه', 'واریته', 'سن',
+        'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3', 'ایستگاه 4', 'ایستگاه 5', 'ارتفاع هفته جاری مزرعه',
+        'ارتفاع هفته گذشته مزرعه', 'رشد هفته جاری', 'رشد هفته گذشته', 'نیتروژن فعلی', 'نیتروژن استاندارد فعلی',
+        'نیتروژن قبلی', 'نیتروژن استاندارد قبلی', 'رطوبت غلاف فعلی', 'رطوبت استاندارد فعلی', 'رطوبت غلاف قبلی',
+        'رطوبت استاندارد قبلی', 'چاهک 1', 'تاریخ قرائت', 'چاهک 2', 'تاریخ قرائت2'
     ])
 
 # Main header
@@ -448,9 +450,9 @@ elif selected == "ورود اطلاعات":
         # Initialize an empty DataFrame for manual entry if not already present
         if 'manual_entry_df' not in st.session_state:
             st.session_state.manual_entry_df = pd.DataFrame(columns=[
-                'ردیف', 'هفته', 'تاریخ قرائت', 'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3',
-                'ایستگاه 4', 'ایستگاه 5', 'ارتفاع هفته جاری مزرعه', 'چاهک 1', 'چاهک 2',
-                'رطوبت غلاف فعلی', 'نیتروژن فعلی', 'واریته', 'سن', 'مساحت داشت', 'کانال', 'اداره'
+                'ردیف', 'نماینده', 'کانال', 'اداره', 'تولید', 'مساحت داشت', 'مساحت زیرمجموعه', 'واریته', 'سن',
+                'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3', 'ایستگاه 4', 'ایستگاه 5', 'ارتفاع هفته جاری مزرعه',
+                'چاهک 1', 'چاهک 2', 'تاریخ قرائت', 'تاریخ قرائت2'
             ])
         
         # Use data_editor for interactive table entry
@@ -459,8 +461,14 @@ elif selected == "ورود اطلاعات":
             num_rows="dynamic",  # Allow dynamic rows
             column_config={
                 'ردیف': st.column_config.TextColumn("ردیف", required=True),
-                'هفته': st.column_config.NumberColumn("هفته", min_value=1, max_value=52, step=1, required=True),
-                'تاریخ قرائت': st.column_config.DateColumn("تاریخ قرائت", format="YYYY-MM-DD", required=True),
+                'نماینده': st.column_config.TextColumn("نماینده", required=True),
+                'کانال': st.column_config.NumberColumn("کانال", min_value=1, step=1, required=True),
+                'اداره': st.column_config.NumberColumn("اداره", min_value=1, step=1, required=True),
+                'تولید': st.column_config.NumberColumn("تولید", min_value=1, step=1, required=True),
+                'مساحت داشت': st.column_config.NumberColumn("مساحت داشت", min_value=0.0, step=0.1, required=True),
+                'مساحت زیرمجموعه': st.column_config.NumberColumn("مساحت زیرمجموعه", min_value=0.0, step=0.1, required=True),
+                'واریته': st.column_config.TextColumn("واریته", required=True),
+                'سن': st.column_config.TextColumn("سن", required=True),
                 'ایستگاه 1': st.column_config.NumberColumn("ایستگاه 1", min_value=0.0, step=0.1, required=True),
                 'ایستگاه 2': st.column_config.NumberColumn("ایستگاه 2", min_value=0.0, step=0.1, required=True),
                 'ایستگاه 3': st.column_config.NumberColumn("ایستگاه 3", min_value=0.0, step=0.1, required=True),
@@ -469,13 +477,8 @@ elif selected == "ورود اطلاعات":
                 'ارتفاع هفته جاری مزرعه': st.column_config.NumberColumn("ارتفاع هفته جاری مزرعه", min_value=0.0, step=0.1, required=True),
                 'چاهک 1': st.column_config.NumberColumn("چاهک 1", min_value=0.0, step=0.1, required=True),
                 'چاهک 2': st.column_config.NumberColumn("چاهک 2", min_value=0.0, step=0.1, required=True),
-                'رطوبت غلاف فعلی': st.column_config.NumberColumn("رطوبت غلاف فعلی", min_value=0.0, max_value=100.0, step=0.1, required=True),
-                'نیتروژن ��علی': st.column_config.NumberColumn("نیتروژن فعلی", min_value=0.0, step=0.1, required=True),
-                'واریته': st.column_config.TextColumn("واریته", required=True),
-                'سن': st.column_config.TextColumn("سن", required=True),
-                'مساحت داشت': st.column_config.NumberColumn("مساحت داشت", min_value=0.0, step=0.1, required=True),
-                'کانال': st.column_config.NumberColumn("کانال", min_value=0, step=1, required=True),
-                'اداره': st.column_config.NumberColumn("اداره", min_value=0, step=1, required=True),
+                'تاریخ قرائت': st.column_config.DateColumn("تاریخ قرائت", format="YYYY-MM-DD", required=True),
+                'تاریخ قرائت2': st.column_config.DateColumn("تاریخ قرائت2", format="YYYY-MM-DD", required=True),
             },
             use_container_width=True,
             hide_index=True
@@ -505,23 +508,52 @@ elif selected == "ورود اطلاعات":
             try:
                 df = pd.read_excel(uploaded_file)
                 
-                # Column mapping based on sample file
+                # Dynamic column mapping based on your file structure
                 column_mapping = {
-                    'Unnamed: 0': 'ردیف', 'هفته': 'هفته', 'تاریخ قرائت': 'تاریخ قرائت',
-                    'ارتفاع هفته جاری مزرعه': 'ارتفاع هفته جاری مزرعه', 'ایستگاه 1': 'ایستگاه 1', 'ایستگاه 2': 'ایستگاه 2',
-                    'ایستگاه 3': 'ایستگاه 3', 'ایستگاه 4': 'ایستگاه 4', 'ایستگاه 5': 'ایستگاه 5',
-                    'چاهک 1': 'چاهک 1', 'چاهک 2': 'چاهک 2', 'رطوبت غلاف فعلی': 'رطوبت غلاف فعلی',
-                    'نیتروژن فعلی': 'نیتروژن فعلی', 'واریته': 'واریته', 'سن': 'سن', 'مساحت داشت': 'مساحت داشت',
-                    'کانال': 'کانال', 'اداره': 'اداره'
+                    'ردیف': 'ردیف',
+                    'نماینده': 'نماینده',
+                    'کانال': 'کانال',
+                    'اداره': 'اداره',
+                    'تولید': 'تولید',
+                    'مساحت داشت': 'مساحت داشت',
+                    'مساحت زیرمجموعه': 'مساحت زیرمجموعه',
+                    'واریته': 'واریته',
+                    'سن': 'سن',
+                    'ایستگاه 1': 'ایستگاه 1',
+                    'ایستگاه 2': 'ایستگاه 2',
+                    'ایستگاه 3': 'ایستگاه 3',
+                    'ایستگاه 4': 'ایستگاه 4',
+                    'ایستگاه 5': 'ایستگاه 5',
+                    'ارتفاع هفته جاری مزرعه': 'ارتفاع هفته جاری مزرعه',
+                    'ارتفاع هفته گذشته مزرعه': 'ارتفاع هفته گذشته مزرعه',
+                    'رشد هفته جاری': 'رشد هفته جاری',
+                    'رشد هفته گذشته': 'رشد هفته گذشته',
+                    'نیتروژن فعلی': 'نیتروژن فعلی',
+                    'نیتروژن استاندارد فعلی': 'نیتروژن استاندارد فعلی',
+                    'نیتروژن قبلی': 'نیتروژن قبلی',
+                    'نیتروژن استاندارد قبلی': 'نیتروژن استاندارد قبلی',
+                    'رطوبت غلاف فعلی': 'رطوبت غلاف فعلی',
+                    'رطوبت استاندارد فعلی': 'رطوبت استاندارد فعلی',
+                    'رطوبت غلاف قبلی': 'رطوبت غلاف قبلی',
+                    'رطوبت استاندارد قبلی': 'رطوبت استاندارد قبلی',
+                    'چاهک 1': 'چاهک 1',
+                    'تاریخ قرائت': 'تاریخ قرائت',
+                    'چاهک 2': 'چاهک 2',
+                    'تاریخ قرائت2': 'تاریخ قرائت2'
                 }
                 
-                df = df.rename(columns=column_mapping)
+                # Map columns dynamically
+                available_columns = df.columns.tolist()
+                mapped_df = pd.DataFrame()
+                for english_col, persian_col in column_mapping.items():
+                    if english_col in available_columns:
+                        mapped_df[persian_col] = df[english_col]
                 
-                # Expected columns
-                expected_columns = list(st.session_state.heights_df.columns)
-                
-                # Check for missing columns
-                missing_columns = [col for col in expected_columns if col not in df.columns]
+                # Ensure all required columns are present
+                required_columns = ['ردیف', 'نماینده', 'کانال', 'اداره', 'تولید', 'مساحت داشت', 'مساحت زیرمجموعه', 'واریته', 'سن',
+                                   'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3', 'ایستگاه 4', 'ایستگاه 5', 'ارتفاع هفته جاری مزرعه',
+                                   'چاهک 1', 'چاهک 2', 'تاریخ قرائت', 'تاریخ قرائت2']
+                missing_columns = [col for col in required_columns if col not in mapped_df.columns]
                 if missing_columns:
                     st.error(f"ستون‌های زیر در فایل یافت نشدند: {', '.join(missing_columns)}")
                 else:
@@ -536,27 +568,25 @@ elif selected == "ورود اطلاعات":
                         except:
                             return date_str
                     
-                    df['تاریخ قرائت'] = df['تاریخ قرائت'].apply(convert_persian_date)
-                    df['تاریخ قرائت'] = pd.to_datetime(df['تاریخ قرائت'], errors='coerce')
+                    mapped_df['تاریخ قرائت'] = mapped_df['تاریخ قرائت'].apply(convert_persian_date)
+                    mapped_df['تاریخ قرائت2'] = mapped_df['تاریخ قرائت2'].apply(convert_persian_date)
+                    mapped_df['تاریخ قرائت'] = pd.to_datetime(mapped_df['تاریخ قرائت'], errors='coerce')
+                    mapped_df['تاریخ قرائت2'] = pd.to_datetime(mapped_df['تاریخ قرائت2'], errors='coerce')
                     
                     # Ensure numeric columns
-                    numeric_cols = ['هفته', 'ارتفاع هفته جاری مزرعه', 'ایستگاه 1', 'ایستگاه 2', 'ایستگاه 3', 
-                                    'ایستگاه 4', 'ایستگاه 5', 'چاهک 1', 'چاهک 2', 'رطوبت غلاف فعلی', 
-                                    'نیتروژن فعلی', 'مساحت داشت', 'کانال', 'اداره']
+                    numeric_cols = ['کانال', 'اداره', 'تولید', 'مساحت داشت', 'مساحت زیرمجموعه', 'ایستگاه 1', 'ایستگاه 2',
+                                    'ایستگاه 3', 'ایستگاه 4', 'ایستگاه 5', 'ارتفاع هفته جاری مزرعه', 'چاهک 1', 'چاهک 2']
                     for col in numeric_cols:
-                        df[col] = pd.to_numeric(df[col], errors='coerce')
-                    
-                    # Select only expected columns
-                    df = df[expected_columns]
+                        mapped_df[col] = pd.to_numeric(mapped_df[col], errors='coerce')
                     
                     # Calculate average height if stations are provided
-                    for index, row in df.iterrows():
+                    for index, row in mapped_df.iterrows():
                         stations = [row[f'ایستگاه {i}'] for i in range(1, 6) if pd.notna(row[f'ایستگاه {i}'])]
                         if stations:
-                            df.loc[index, 'ارتفاع هفته جاری مزرعه'] = round(sum(stations) / len(stations), 1)
+                            mapped_df.loc[index, 'ارتفاع هفته جاری مزرعه'] = round(sum(stations) / len(stations), 1)
                     
                     # Append to session state
-                    st.session_state.heights_df = pd.concat([st.session_state.heights_df, df], ignore_index=True)
+                    st.session_state.heights_df = pd.concat([st.session_state.heights_df, mapped_df], ignore_index=True)
                     st.success("داده‌ها از فایل با موفقیت بارگذاری شدند!")
                     st.dataframe(st.session_state.heights_df.tail())
             except Exception as e: