Update app.py

app.py

@@ -15,13 +15,13 @@ def _require_client(client_obj):
 
 def _progress_html(pct: float, text: str) -> str:
     pct = max(0, min(100, pct))
-    return f\"\"\"
-    <div class="progress-card">
-      <div class="progress-head">{text}</div>
-      <div class="pbar"><div class="pbar-fill" style="width:{pct:.0f}%"></div></div>
-      <div class="pbar-foot">{pct:.0f}%</div>
-    </div>
-    \"\"\"
+    return (
+        "<div class=\"progress-card\">"
+        f"<div class=\"progress-head\">{text}</div>"
+        f"<div class=\"pbar\"><div class=\"pbar-fill\" style=\"width:{pct:.0f}%\"></div></div>"
+        f"<div class=\"pbar-foot\">{pct:.0f}%</div>"
+        "</div>"
+    )
 
 def ui_set_api_key(api_key: str):
     api_key = (api_key or "").strip()
@@ -33,8 +33,7 @@ def ui_set_api_key(api_key: str):
     except Exception as e:
         return None, f"❌ Failed to set API key: {e}"
 
-def upload_and_index(client_state, file_obj,
-                     progress=gr.Progress(track_tqdm=True)):
+def upload_and_index(client_state, file_obj, progress=gr.Progress(track_tqdm=True)):
     if client_state is None:
         yield None, "❌ Set API key first.", _progress_html(0, "Waiting for API key")
         return
@@ -44,7 +43,8 @@ def upload_and_index(client_state, file_obj,
 
     client = _require_client(client_state)
 
-    store = client.file_search_stores.create(config={"display_name": "my-upload-store"})
+    # Create a new store
+    store = client.file_search_stores.create(config={"display_name": "uploaded-store"})
     store_name = store.name
 
     fname = Path(file_obj.name).name
@@ -74,16 +74,16 @@ def upload_and_index(client_state, file_obj,
 
     yield store_name, f"✅ File indexed into store `{store_name}`", _progress_html(100, "Done")
 
-def ask(client_state, store_name: str, history_msgs, question: str, model_id: str):
+def ask(client_state, store_name: str, history, question: str, model_id: str):
     if client_state is None:
-        return history_msgs, "❌ Set API key first."
+        return history, "❌ Set API key first."
     if not store_name:
-        return history_msgs, "⚠️ Upload & index a file first."
-
-    client = _require_client(client_state)
+        return history, "⚠️ Upload & index a file first."
     q = (question or "").strip()
     if not q:
-        return history_msgs, "⚠️ Please type a question."
+        return history, "⚠️ Please type a question."
+
+    client = _require_client(client_state)
 
     tool = types.Tool(
         file_search=types.FileSearch(
@@ -96,13 +96,15 @@ def ask(client_state, store_name: str, history_msgs, question: str, model_id: st
         config=types.GenerateContentConfig(tools=[tool]),
     )
     answer = resp.text or "No answer."
-    history = list(history_msgs or [])
+    history = history or []
     history.append({"role": "user", "content": q})
     history.append({"role": "assistant", "content": answer})
 
-    return history, history  # return history as both chat history & output
+    return history, history
 
-custom_css = \"\"\"/* Optional CSS here */\"\"\"
+custom_css = """
+/* Optional: add your custom CSS here */
+"""
 
 with gr.Blocks(title="Gemini File Search – Upload & Chat Demo") as demo:
     gr.HTML(f"<style>{custom_css}</style>")
@@ -125,7 +127,7 @@ with gr.Blocks(title="Gemini File Search – Upload & Chat Demo") as demo:
     gr.Markdown("### Ask questions about the uploaded file")
     question_tb = gr.Textbox(placeholder="Type your question…", show_label=False)
     ask_btn = gr.Button("Ask")
-    chatbot = gr.Chatbot()  # No `type=` argument
+    chatbot = gr.Chatbot()
 
     api_btn.click(ui_set_api_key, [api_tb], [client_state, api_status])