Update app.py

app.py

@@ -1,7 +1,9 @@
 import os
-import base64
 import gradio as gr
 import openai
+
+# Optional imports for extended functionality
+import base64
 import fitz  # PyMuPDF
 from io import BytesIO
 from moviepy.video.io.VideoFileClip import VideoFileClip
@@ -23,12 +25,12 @@ def save_api_key(key: str) -> str:
 
 def get_api_key(input_key: str) -> str:
     """Determine which OpenAI key to use: input > saved > env."""
-    # 1️⃣ User-provided
+    # 1️⃣ User-provided key
     if input_key and input_key.strip():
         save_api_key(input_key)
         return input_key.strip()
 
-    # 2️⃣ Previously saved
+    # 2️⃣ Previously saved key
     if os.path.exists(KEY_FILE):
         with open(KEY_FILE) as f:
             return f.read().strip()
@@ -39,14 +41,14 @@ def get_api_key(input_key: str) -> str:
         save_api_key(env_key)
         return env_key
 
-    # ❗ None found
+    # ❗ No key found
     raise gr.Error("❗ OpenAI API key required.")
 
 
 def apply_filters(files, filters):
     """
     Filter uploaded files by extension and user-selected filters.
-    Returns a list for the Gallery display and a list of (path, type) for tracking.
+    Returns a list for the Gallery display and tracking info for table generation.
     """
     gallery_items = []
     tracking = []
@@ -55,7 +57,6 @@ def apply_filters(files, filters):
         return gallery_items, tracking
 
     for f in files:
-        # file may be a tempfile object or filepath string
         path = getattr(f, "name", f)
         ext = os.path.splitext(path)[1].lower()
 
@@ -82,7 +83,6 @@ def generate_table(selected, api_key):
     key = get_api_key(api_key)
     openai.api_key = key
 
-    # Header
     md = "| ✅ | Type | Filename |\n|---|------|----------|\n"
     emoji_map = {
         'image': '🖼️',
@@ -117,30 +117,27 @@ def chat_handler(api_key, message, history):
     history.append((message, answer))
     return history
 
-
 # 🔑 UI Definition
 with gr.Blocks(title="🔬🧠 ScienceBrain.Gradio") as demo:
     gr.Markdown("# 🔬🧠 ScienceBrain Gradio")
 
+    # --- API Key Input & Save ---
     with gr.Row():
         api_input = gr.Textbox(
-            label="🔑 OpenAI Key",
-            type="password",
-            value=os.getenv("OPENAI_KEY", "")
+            label="🔑 OpenAI Key", type="password", value=os.getenv("OPENAI_KEY", "")
         )
         save_btn = gr.Button("💾 Save Key")
         status_txt = gr.Textbox(interactive=False)
-
         save_btn.click(save_api_key, inputs=api_input, outputs=status_txt)
 
-    # 📋 Media Gallery & Filters
+    # --- Media Gallery & Filters ---
     gr.Markdown("## 📋 Media Gallery & Filters")
     upload = gr.File(file_count="multiple", label="Upload files (images, audio, videos, PDFs)")
     gallery = gr.Gallery(label="Filtered Gallery", columns=4)
     filter_opts = ["🖼️ Images", "🎤 Audio", "🎥 Video", "📄 PDF"]
     filters = gr.CheckboxGroup(filter_opts, value=filter_opts, label="🔍 Filter Types")
     select_btn = gr.Button("⚙️ Apply Filters")
-    selected = gr.Variable()
+    selected = gr.State()
 
     select_btn.click(
         apply_filters,
@@ -148,26 +145,17 @@ with gr.Blocks(title="🔬🧠 ScienceBrain.Gradio") as demo:
         outputs=[gallery, selected]
     )
 
-    # ✅ Include in Discussion
-    gr.Markdown("## ✅ Include in Discussion")
-    disc = gr.CheckboxGroup(label="Select items", choices=[])  
-    gallery.select(
-        lambda x: [x],
-        inputs=[gallery],
-        outputs=[disc]
-    )
-
-    # 📝 Summary Table
+    # --- Summary Table ---
     gr.Markdown("## 📝 Summary Table")
     table_md = gr.Markdown()
     table_btn = gr.Button("Generate Table")
     table_btn.click(
         generate_table,
-        inputs=[disc, api_input],
+        inputs=[selected, api_input],
         outputs=[table_md]
     )
 
-    # 💬 Chat Tab
+    # --- Chat Tab ---
     with gr.Tab("💬 Chat"):
         chatbot = gr.Chatbot()
         msg_in = gr.Textbox(placeholder="Type your message...")