Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import assemblyai as aai
 import gradio as gr
 import requests
+import time
 from supabase import create_client, Client
 
 # --------------------------------------------------
@@ -29,7 +30,54 @@ supabase: Client = create_client(SUPABASE_URL, SUPABASE_KEY)
 # --------------------------------------------------
 # In-memory session
 # --------------------------------------------------
-session = {"user": None, "user_id": None}
+INACTIVITY_TIMEOUT = 30 * 60  # 30 minutes in seconds
+session = {"user": None, "user_id": None, "last_activity": None}
+
+# --------------------------------------------------
+# Check for existing Supabase session on startup
+# --------------------------------------------------
+def restore_session():
+    try:
+        user = supabase.auth.get_user()
+        if user and user.user:
+            session["user"] = user.user.email
+            session["user_id"] = user.user.id
+            return True
+    except Exception as e:
+        print(f"⚠️ Session restore error: {e}")
+    return False
+
+# Restore session on app startup
+restore_session()
+
+# --------------------------------------------------
+# Inactivity timeout check
+# --------------------------------------------------
+def check_inactivity():
+    """Check if user has been inactive for 30 mins, auto-logout if true"""
+    if session["user_id"] is None:
+        return False
+    
+    if session["last_activity"] is None:
+        return False
+    
+    elapsed = time.time() - session["last_activity"]
+    if elapsed > INACTIVITY_TIMEOUT:
+        # Auto logout
+        try:
+            supabase.auth.sign_out()
+        except:
+            pass
+        session["user"] = None
+        session["user_id"] = None
+        session["last_activity"] = None
+        return True  # Timed out
+    
+    return False
+
+def update_activity():
+    """Update last activity timestamp"""
+    session["last_activity"] = time.time()
 
 # --------------------------------------------------
 # Auth functions
@@ -40,6 +88,7 @@ def signup(email, password):
         if res.user:
             session["user"] = res.user.email
             session["user_id"] = res.user.id
+            session["last_activity"] = time.time()
             return (
                 "✅ Signup successful.",
                 gr.update(visible=False),  # hide auth
@@ -55,6 +104,7 @@ def login(email, password):
         if res.user:
             session["user"] = res.user.email
             session["user_id"] = res.user.id
+            session["last_activity"] = time.time()
             return (
                 "✅ Login successful.",
                 gr.update(visible=False),  # hide auth
@@ -64,6 +114,19 @@ def login(email, password):
     except Exception as e:
         return f"❌ Login error: {str(e)}", gr.update(), gr.update()
 
+def logout():
+    try:
+        supabase.auth.sign_out()
+        session["user"] = None
+        session["user_id"] = None
+        return (
+            "✅ Logged out successfully.",
+            gr.update(visible=True),   # show auth
+            gr.update(visible=False)   # hide transcription
+        )
+    except Exception as e:
+        return f"❌ Logout error: {str(e)}", gr.update(), gr.update()
+
 # --------------------------------------------------
 # Transcription logic
 # --------------------------------------------------
@@ -77,6 +140,10 @@ def transcribe(audio_path: str):
     return transcript.text, transcript.audio_duration
 
 def transcribe_audio(audio_file):
+    # Check for inactivity timeout
+    if check_inactivity():
+        return "❌ Session expired due to inactivity. Please log in again."
+    
     if session["user_id"] is None:
         return "❌ Please log in."
 
@@ -84,6 +151,9 @@ def transcribe_audio(audio_file):
         return "⚠️ Please upload an audio file."
 
     try:
+        # Update activity timestamp
+        update_activity()
+        
         # Transcribe audio
         text, duration_seconds = transcribe(audio_file)
         minutes_used = max(1, int(duration_seconds // 60))
@@ -91,6 +161,7 @@ def transcribe_audio(audio_file):
         # Save transcription
         supabase.table("transcriptions").insert({
             "user_id": session["user_id"],
+            "user_email": session["user"],
             "audio_duration": minutes_used,
             "transcript": text
         }).execute()
@@ -130,7 +201,7 @@ def transcribe_audio(audio_file):
 with gr.Blocks() as app:
 
     # 🔐 AUTH SECTION
-    with gr.Column(visible=True) as auth_section:
+    with gr.Column(visible=not restore_session()) as auth_section:
         gr.Markdown("## 🔐 Login or Signup")
         email = gr.Textbox(label="Email")
         password = gr.Textbox(label="Password", type="password")
@@ -141,9 +212,13 @@ with gr.Blocks() as app:
             signup_btn = gr.Button("Signup")
 
     # 🎙️ TRANSCRIPTION SECTION (hidden initially)
-    with gr.Column(visible=False) as transcription_section:
+    with gr.Column(visible=restore_session()) as transcription_section:
         gr.Markdown("## 🎙️ AI Transcription")
         gr.Markdown("Upload audio → get transcript")
+        
+        with gr.Row():
+            logout_btn = gr.Button("🚪 Logout", scale=1)
+        
         audio_input = gr.Audio(type="filepath", label="Upload Audio")
         output_text = gr.Textbox(label="Transcript", lines=10)
         transcribe_btn = gr.Button("Transcribe")
@@ -153,8 +228,12 @@ with gr.Blocks() as app:
         gr.Markdown("## 📋 Site Report Form")
         gr.HTML(
             '<iframe src="http://13.48.132.18:5678/form/SITEREPORT" '
-            'width="100%" height="700" frameborder="0" style="border: 1px solid #ccc; border-radius: 5px;"></iframe>'
+            'width="100%" height="700" frameborder="0" '
+            'allow="same-origin" allowfullscreen '
+            'style="border: 1px solid #ccc; border-radius: 5px; background: white;"></iframe>'
         )
+        
+        logout_status = gr.Textbox(label="Status", visible=False)
 
     # --------------------------------------------------
     # Button bindings
@@ -168,6 +247,11 @@ with gr.Blocks() as app:
         [email, password],
         [auth_status, auth_section, transcription_section]
     )
+    logout_btn.click(
+        logout,
+        None,
+        [logout_status, transcription_section, auth_section]
+    )
 
 # --------------------------------------------------
 # Launch