Update app.py

app.py

@@ -8,7 +8,7 @@ from google.genai import types
 MODEL_NAME = "gemini-2.5-flash"
 embedder = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
 
-# ✅ Load knowledge base from file (you create knowledge.txt in the repo)
+# ✅ Load knowledge base from file (put your content inside knowledge.txt in repo)
 with open("knowledge.txt", "r", encoding="utf-8") as f:
     raw_text = f.read()
 
@@ -39,7 +39,6 @@ If the answer is not in the context, reply:
 """
 USER_TEMPLATE = """Client asked:
 {question}
-
 Relevant chat context:
 {context}
 """
@@ -54,7 +53,7 @@ def retrieve(query, top_k=3):
         results.append(meta[i]["chunk"])
     return results
 
-# ✅ Call Gemini (with user-provided API key)
+# ✅ Call Gemini
 def call_gemini(api_key, system_prompt, user_prompt):
     try:
         client = genai.Client(api_key=api_key)
@@ -80,24 +79,32 @@ def answer_question(api_key, question, top_k=3):
     user_prompt = USER_TEMPLATE.format(question=question, context=context)
     return call_gemini(api_key, SYSTEM_TEMPLATE, user_prompt)
 
-# ✅ Gradio Chatbot
-def chatbot_fn(message, history, api_key):
-    reply = answer_question(api_key, message)
-    return reply
-
+# ✅ Gradio App
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 Arduino Fiverr Chatbot (Gemini Powered)")
 
-    with gr.Row():
-        api_key_box = gr.Textbox(
-            label="🔑 Enter your Gemini API Key",
-            placeholder="Paste your Gemini API key here...",
-            type="password",
-        )
-
-    gr.ChatInterface(
-        fn=lambda message, history: chatbot_fn(message, history, api_key_box.value),
-        title="Arduino Client Assistant"
+    api_key_box = gr.Textbox(
+        label="🔑 Enter your Gemini API Key",
+        placeholder="Paste your Gemini API key here...",
+        type="password",
     )
 
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Ask your Arduino client question here...")
+    state = gr.State("")
+
+    def save_key(api_key):
+        return api_key, "✅ API Key saved! You can now ask questions."
+
+    def chatbot_fn(message, history, api_key):
+        reply = answer_question(api_key, message)
+        history.append((message, reply))
+        return history, ""
+
+    # Save API key
+    api_key_box.change(save_key, inputs=[api_key_box], outputs=[state, chatbot])
+
+    # Chat flow
+    msg.submit(chatbot_fn, inputs=[msg, chatbot, state], outputs=[chatbot, msg])
+
 demo.launch()