Update dashboard.py

dashboard.py

@@ -63,82 +63,40 @@ def predict_image(image):
     label = "✅ Real Image" if prediction >= 0.5 else "⚠️ Fake Image"
     return f"{label} (Confidence: {confidence:.2%})"
 
-def register_user(name, phone, email, gender, password):
-    if not all([name, phone, email, gender, password]):
-        return "❌ All fields are required for signup."
-    if not is_valid_email(email): return "❌ Invalid email format."
-    if not is_valid_phone(phone): return "❌ Phone must be 10 digits."
-
-    query_url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?email=eq.{email}"
-    r = requests.get(query_url, headers=headers)
-    if r.status_code == 200 and len(r.json()) > 0:
-        return "⚠️ Email already registered."
-
-    hashed_pw = bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode()
-    data = {
-        "name": name,
-        "phone": phone,
-        "email": email,
-        "gender": gender,
-        "password": hashed_pw
-    }
-    r = requests.post(f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}", headers=headers, data=json.dumps(data))
-    return "✅ Registration successful! Please log in." if r.status_code == 201 else "❌ Error during registration."
-
-def login_user(email, password):
-    url = f"{SUPABASE_URL}/rest/v1/{SUPABASE_TABLE}?email=eq.{email}"
-    r = requests.get(url, headers=headers)
-    if r.status_code == 200 and r.json():
-        stored_hash = r.json()[0]["password"]
-        return bcrypt.checkpw(password.encode(), stored_hash.encode())
-    return False
-
 # --- UI ---
 with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as demo:
-    is_logged_in = gr.State(False)
+    # Custom CSS for centering content
+    gr.HTML("""
+    <style>
+        .home-content {
+            text-align: center;
+            margin-top: 100px;
+        }
+    </style>
+    """)
 
     HOME_TAB_NAME = "🏠 Home"
-    LOGIN_TAB_NAME = "🔐 Login"
     DETECT_TAB_NAME = "🧪 Detect Deepfake"
     ABOUT_TAB_NAME = "ℹ️ About"
     COMMUNITY_TAB_NAME = "🌐 Community"
     GUIDE_TAB_NAME = "📘 User Guide"
 
-    with gr.Tabs(selected=HOME_TAB_NAME) as tabs:
-        # --- HOME TAB ---
+    tabs = gr.Tabs()
+
+    with tabs:
         with gr.Tab(HOME_TAB_NAME) as home_tab:
-            with gr.Column():
-                gr.Markdown("""
-                # 👁️‍🗨️ Welcome to VerifiAI  
-                Your trusted assistant for detecting deepfakes in images using AI.
+            with gr.Row(align="center", justify="center"):
+                with gr.Column():
+                    gr.Markdown("""
+                    # 👁️‍🗨️ Welcome to VerifiAI  
+                    Your trusted assistant for detecting deepfakes in images using AI.
 
-                🔍 Upload images, analyze authenticity, and learn how deepfakes work.
+                    🔍 Upload images, analyze authenticity, and learn how deepfakes work.
 
-                👉 Click the **Login** tab to access the detector.
-                """)
-                go_login_btn = gr.Button("🔐 Go to Login")
+                    👉 Use the tabs above to get started.
+                    """, elem_classes="home-content")
 
-        # --- LOGIN TAB ---
-        with gr.Tab(LOGIN_TAB_NAME) as login_tab:
-            with gr.Row():
-                with gr.Column(scale=1):
-                    gr.Markdown("## Welcome!", "Login to access the detector, or sign up for a new account.")
-                with gr.Column(scale=2):
-                    gr.Markdown("### Login or Sign Up")
-                    message_output = gr.Markdown(visible=False)
-                    email_login = gr.Textbox(label="Email")
-                    password_login = gr.Textbox(label="Password", type="password")
-                    login_btn = gr.Button("Login", variant="primary")
-                    with gr.Accordion("New User? Click here to Sign Up", open=False) as signup_accordion:
-                        name_signup = gr.Textbox(label="Name")
-                        phone_signup = gr.Textbox(label="Phone (10 digits)")
-                        email_signup = gr.Textbox(label="Email")
-                        gender_signup = gr.Dropdown(label="Gender", choices=["Male", "Female", "Other"])
-                        password_signup = gr.Textbox(label="Create Password", type="password")
-                        signup_btn = gr.Button("Sign Up")
-
-        # --- DETECTION TAB ---
-        with gr.Tab(DETECT_TAB_NAME, visible=False) as detect_tab:
+        with gr.Tab(DETECT_TAB_NAME) as detect_tab:
             with gr.Row():
                 gr.Markdown("## Deepfake Detector")
                 logout_btn = gr.Button("Logout")
@@ -148,60 +106,15 @@ with gr.Blocks(theme=gr.themes.Soft(), title="VerifiAI - Deepfake Detector") as
                     result = gr.Textbox(label="Prediction Result", interactive=False)
                     predict_btn = gr.Button("Predict", variant="primary")
 
-        # --- OTHER TABS ---
         with gr.Tab(ABOUT_TAB_NAME): about.layout()
         with gr.Tab(COMMUNITY_TAB_NAME): community.layout()
         with gr.Tab(GUIDE_TAB_NAME): user_guide.layout()
 
-    # --- Logic Handlers ---
-    def update_ui_on_auth_change(logged_in_status):
-        if logged_in_status:
-            return (
-                gr.update(visible=False),
-                gr.update(visible=True),
-                gr.update(selected=GUIDE_TAB_NAME),
-                gr.update(value="✅ Login successful!", visible=True)
-            )
-        else:
-            return (
-                gr.update(visible=True),
-                gr.update(visible=False),
-                gr.update(selected=LOGIN_TAB_NAME),
-                gr.update(value="", visible=False)
-            )
-
-    def handle_login(email, password):
-        if login_user(email, password):
-            return True, gr.update(value="✅ Login successful!", visible=True)
-        else:
-            return False, gr.update(value="❌ Invalid email or password.", visible=True)
-
-    def handle_logout():
-        return False, "", ""
-
-    def handle_signup(name, phone, email, gender, password):
-        msg = register_user(name, phone, email, gender, password)
-        if msg.startswith("✅"):
-            return gr.update(value=msg, visible=True), "", "", "", "", "", gr.update(open=False)
-        else:
-            return gr.update(value=msg, visible=True), name, phone, email, gender, password, gr.update(open=True)
-
-    def go_to_login():
-        return gr.update(selected=LOGIN_TAB_NAME)
-
-    # --- Events ---
-    go_login_btn.click(fn=go_to_login, outputs=tabs)
-    login_btn.click(fn=handle_login, inputs=[email_login, password_login], outputs=[is_logged_in, message_output])
-    logout_btn.click(fn=handle_logout, inputs=[], outputs=[is_logged_in, email_login, password_login])
-    is_logged_in.change(fn=update_ui_on_auth_change, inputs=is_logged_in, outputs=[login_tab, detect_tab, tabs, message_output])
-    signup_btn.click(
-        fn=handle_signup,
-        inputs=[name_signup, phone_signup, email_signup, gender_signup, password_signup],
-        outputs=[message_output, name_signup, phone_signup, email_signup, gender_signup, password_signup, signup_accordion]
-    )
+    # Ensure Home tab is selected on load
+    demo.load(fn=lambda: gr.update(selected=HOME_TAB_NAME), outputs=tabs)
+
     predict_btn.click(fn=predict_image, inputs=image_input, outputs=result)
 
-    demo.load(fn=lambda: False, outputs=is_logged_in)
 
 if __name__ == "__main__":
     demo.launch()