Update app.py

app.py

@@ -74,6 +74,8 @@ if 'authenticated' not in st.session_state:
     st.session_state.authenticated = False
 if 'is_admin' not in st.session_state:
     st.session_state.is_admin = False
+if 'page' not in st.session_state:
+    st.session_state.page = "login"
 
 # Register Page
 def register_page():
@@ -83,16 +85,19 @@ def register_page():
     new_password = st.text_input("رمز عبور جدید", type='password')
 
     if st.button("ثبت نام"):
-        user_details = load_user_details()
+        if new_username and new_password:
+            user_details = load_user_details()
 
-        if new_username in user_details['username'].values:
-            st.error("این نام کاربری قبلاً ثبت شده است.")
+            if new_username in user_details['username'].values:
+                st.error("این نام کاربری قبلاً ثبت شده است.")
+            else:
+                new_user = pd.DataFrame([[new_username, new_password]], columns=['username', 'password'])
+                user_details = pd.concat([user_details, new_user], ignore_index=True)
+                save_user_details(user_details)
+                st.success("ثبت نام با موفقیت انجام شد! اکنون می‌توانید وارد شوید.")
+                st.session_state.page = "login"
         else:
-            new_user = pd.DataFrame([[new_username, new_password]], columns=['username', 'password'])
-            user_details = pd.concat([user_details, new_user], ignore_index=True)
-            save_user_details(user_details)
-            st.success("ثبت نام با موفقیت انجام شد! اکنون می‌توانید وارد شوید.")
-            st.session_state.page = "login"
+            st.error("لطفاً نام کاربری و رمز عبور را وارد کنید.")
 
 # Login Page
 def login_page():
@@ -111,12 +116,14 @@ def login_page():
             st.session_state.is_admin = (st.session_state.username == "armin")
             # Load schedule data again after login
             st.session_state.schedule_data = load_schedule()
+            st.session_state.page = "schedule"
             st.experimental_rerun()
         else:
             st.error("نام کاربری یا رمز عبور اشتباه است. لطفا دوباره تلاش کنید.")
     
     if st.button("ثبت نام"):
         st.session_state.page = "register"
+        st.experimental_rerun()
 
 # Class Schedule Page
 def schedule_page():
@@ -140,7 +147,7 @@ def schedule_page():
         
         if st.button("اضافه کردن کلاس"):
             class_datetime = get_class_datetime(day_selected, start_time_selected)
-            if class_datetime - datetime.now() < timedelta(hours=90):
+            if class_datetime - datetime.now() < timedelta(hours=30):
                 st.error("کلاس‌ها باید حداقل ۳۰ ساعت قبل رزرو شوند.")
             elif not has_system_availability(st.session_state.schedule_data, class_selected, day_selected, start_time_selected, system_selected):
                 st.error(f"سیستم '{system_selected}' در زمان '{start_time_selected}' در روز '{day_selected}' برای '{class_selected}' موجود نیست.")
@@ -193,12 +200,10 @@ def schedule_page():
         st.session_state['username'] = ""
         st.session_state['password'] = ""
         st.session_state['is_admin'] = False
+        st.session_state.page = "login"
         st.experimental_rerun()
 
 # Page navigation
-if 'page' not in st.session_state:
-    st.session_state.page = "login"
-
 if st.session_state.page == "login":
     login_page()
 elif st.session_state.page == "register":