Update dashboard.py

dashboard.py

@@ -12,16 +12,11 @@ import warnings
 
 from pages import about, community, user_guide
 
-
-
-# --- Initial Setup ---
-
-# Suppress warnings and TensorFlow logs
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # Changed to 2 to hide some informational logs
+# --- Setup ---
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
 warnings.filterwarnings("ignore")
 np.seterr(all='ignore')
 
-# Load model (or create a dummy one if not found)
 MODEL_PATH = "model_15_64.h5"
 if not os.path.exists(MODEL_PATH):
     print(f"Model file '{MODEL_PATH}' not found. Creating a dummy model for testing.")
@@ -30,18 +25,15 @@ if not os.path.exists(MODEL_PATH):
         tf.keras.layers.Flatten(),
         tf.keras.layers.Dense(1, activation='sigmoid')
     ])
-    # Suppress the warning during saving
     with warnings.catch_warnings():
         warnings.simplefilter("ignore")
         dummy_model.save(MODEL_PATH)
 
-# Suppress the "Compiled the loaded model" warning from Keras
 with warnings.catch_warnings():
     warnings.simplefilter("ignore")
     deepfake_model = tf.keras.models.load_model(MODEL_PATH)
 
-
-# Database setup
+# --- DB ---
 DB_PATH = os.path.abspath("users.db")
 conn = sqlite3.connect(DB_PATH, check_same_thread=False)
 cursor = conn.cursor()
@@ -58,16 +50,13 @@ CREATE TABLE IF NOT EXISTS user_details (
 conn.commit()
 conn.close()
 
-# --- Helper Functions ---
-
-# Validators
+# --- Helpers ---
 def is_valid_email(email):
     return re.match(r"[^@]+@[^@]+\.[^@]+", email)
 
 def is_valid_phone(phone):
     return re.match(r"^[0-9]{10}$", phone)
 
-# Image preprocessing
 def preprocess_image(image):
     if image.mode != 'RGB':
         image = image.convert('RGB')
@@ -85,7 +74,6 @@ def predict_image(image):
     label = "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
     return f"{label} (Confidence: {confidence:.2%})"
 
-# Auth logic
 def register_user(name, phone, email, password):
     if not all([name, phone, email, password]):
         return "❌ All fields are required for signup."
@@ -133,7 +121,7 @@ def login_user(email, password):
         if conn:
             conn.close()
 
-# --- Gradio UI ---
+# --- UI ---
 with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
     is_logged_in = gr.State(False)
 
@@ -145,8 +133,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
 
     with gr.Tabs(selected=LOGIN_TAB_NAME) as tabs:
         with gr.Tab(LOGIN_TAB_NAME) as login_tab:
-            # --- THIS IS THE FIX ---
-            # The .style() method was removed. The row will work correctly without it.
             with gr.Row():
                 with gr.Column(scale=1):
                     gr.Markdown("## Welcome!", "Login to access the detector, or sign up for a new account.")
@@ -157,13 +143,14 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
                     email_login = gr.Textbox(label="Email", elem_id="login_email")
                     password_login = gr.Textbox(label="Password", type="password", elem_id="login_pass")
                     login_btn = gr.Button("Login", variant="primary")
-                    
+
                     with gr.Accordion("New User? Click here to Sign Up", open=False):
-                        name_signup = gr.Textbox(label="Name")
-                        phone_signup = gr.Textbox(label="Phone (10 digits)")
-                        email_signup = gr.Textbox(label="Email")
-                        password_signup = gr.Textbox(label="Create Password", type="password")
-                        signup_btn = gr.Button("Sign Up")
+                        with gr.Group(visible=True) as signup_group:
+                            name_signup = gr.Textbox(label="Name")
+                            phone_signup = gr.Textbox(label="Phone (10 digits)")
+                            email_signup = gr.Textbox(label="Email")
+                            password_signup = gr.Textbox(label="Create Password", type="password")
+                            signup_btn = gr.Button("Sign Up")
 
         with gr.Tab(DETECT_TAB_NAME, visible=False) as detect_tab:
             with gr.Row():
@@ -181,7 +168,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
         with gr.Tab(COMMUNITY_TAB_NAME): community.layout()
         with gr.Tab(GUIDE_TAB_NAME): user_guide.layout()
 
-    # --- Backend Logic Handlers ---
+    # --- Events ---
     def handle_login(email, password):
         if login_user(email, password):
             return (
@@ -200,44 +187,46 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
         if msg.startswith("✅"):
             return (
                 gr.update(value=msg, visible=True),
-                "", "", "", "",  # clear signup fields
-                gr.update(open=False)  # close accordion
+                "", "", "", "",  # clear inputs
+                gr.update(visible=False)  # hide signup form
             )
         else:
             return (
                 gr.update(value=msg, visible=True),
-                name, phone, email, password,  # keep values if error
-                gr.update(open=True)  # keep accordion open
+                name, phone, email, password,
+                gr.update(visible=True)
             )
 
     def handle_logout():
         return (
-            False, gr.update(visible=True), gr.update(visible=False),
-            gr.update(selected=LOGIN_TAB_NAME), "", ""
+            False,
+            gr.update(visible=True), gr.update(visible=False),
+            gr.update(selected=LOGIN_TAB_NAME),
+            "", ""
         )
 
-    # --- Event Listeners ---
     login_btn.click(
         fn=handle_login,
         inputs=[email_login, password_login],
         outputs=[message_output, is_logged_in, login_tab, detect_tab, tabs]
     )
-    
+
     signup_btn.click(
-    fn=handle_signup,
-    inputs=[name_signup, phone_signup, email_signup, password_signup],
-    outputs=[
-        message_output,
-        name_signup, phone_signup, email_signup, password_signup,
-        gr.Accordion.update()]
+        fn=handle_signup,
+        inputs=[name_signup, phone_signup, email_signup, password_signup],
+        outputs=[
+            message_output,
+            name_signup, phone_signup, email_signup, password_signup,
+            signup_group
+        ]
     )
 
-  
     logout_btn.click(
         fn=handle_logout,
         inputs=[],
         outputs=[is_logged_in, login_tab, detect_tab, tabs, email_login, password_login]
     )
+
     predict_btn.click(
         fn=predict_image,
         inputs=image_input,
@@ -245,4 +234,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Deepfake Detector") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()