Update app.py

app.py

@@ -7,17 +7,18 @@ import json
 from sentence_transformers import SentenceTransformer
 from datetime import datetime
 
-# ✅ Load context and model
+# ✅ Load RAG context
 with open("texts.json", "r", encoding="utf-8") as f:
     texts = json.load(f)
 
 index = faiss.read_index("faiss_index.bin")
 embed_model = SentenceTransformer("all-MiniLM-L6-v2")
 
+# ✅ API setup
 API_KEY = os.environ.get("OPENROUTER_API_KEY")
 MODEL = "qwen/qwen-2.5-coder-32b-instruct:free"
 
-# ✅ Get context
+# ✅ Get relevant context
 def get_context(query, top_k=5, threshold=0.45):
     query_vec = embed_model.encode([query])
     D, I = index.search(np.array(query_vec), top_k)
@@ -29,7 +30,7 @@ def get_context(query, top_k=5, threshold=0.45):
 def chat_fn(message, history):
     context = get_context(message)
     if context is None:
-        return history + [(message, "❌ Sorry! I cannot answer that.")], gr.update(visible=True)
+        return history[:-1] + [(message, "❌ Sorry! I cannot answer that.")], gr.update(visible=True)
 
     headers = {
         "Authorization": f"Bearer {API_KEY}",
@@ -43,7 +44,7 @@ def chat_fn(message, history):
         }
     ]
 
-    for user, assistant in history:
+    for user, assistant in history[:-1]:
         messages.append({"role": "user", "content": user})
         messages.append({"role": "assistant", "content": assistant})
 
@@ -61,9 +62,9 @@ def chat_fn(message, history):
     except Exception as e:
         reply = f"❌ Error: {e}"
 
-    return history + [(message, reply)], gr.update(visible=True)
+    return history[:-1] + [(message, reply)], gr.update(visible=True)
 
-# ✅ Export chat
+# ✅ Export logs to file
 def export_logs(history):
     timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
     filename = f"chat_log_{timestamp}.txt"
@@ -72,18 +73,20 @@ def export_logs(history):
         f.write(log_text)
     return filename
 
-# ✅ Gradio UI with typing animation + auto-scroll + export button
-with gr.Blocks(css=".footer {display: none !important;}") as demo:
-    chatbot = gr.Chatbot()
+# ✅ UI
+with gr.Blocks(css="""
+  .footer {display: none !important;}
+  #chat-window {height: 500px !important; overflow-y: auto;}
+""") as demo:
+    chatbot = gr.Chatbot(elem_id="chat-window")
     state = gr.State([])
+
     with gr.Row():
         msg = gr.Textbox(placeholder="Type your message and press enter...", scale=8)
         export_btn = gr.Button("Export Chat", scale=1, visible=False)
 
-    typing_indicator = gr.Textbox(value="", visible=False, interactive=False, show_label=False)
-
+    # Typing simulation
     def user_send(message, history):
-        chatbot.update(value=history + [(message, "⏳ ...")])  # Typing animation
         return "", history + [(message, "⏳ ...")], gr.update(visible=False)
 
     def complete_chat(message, history):
@@ -95,6 +98,4 @@ with gr.Blocks(css=".footer {display: none !important;}") as demo:
 
     export_btn.click(fn=export_logs, inputs=[state], outputs=gr.File(label="Download Chat Log"))
 
-    chatbot.style(height=500)
-
 demo.launch()