Update app.py

app.py

@@ -18,12 +18,20 @@ def set_api_key(api_key):
 # --- Chat function ---
 def chat(user_input, history):
     global knowledge_base, global_api_key
+    history = history or []
+
     if not global_api_key:
         return history, "⚠️ Please set your Gemini API key first."
 
     if not knowledge_base:
         return history, "⚠️ Knowledge base is empty."
 
+    # Combine knowledge base + chat history + current input
+    context_text = f"Knowledge base:\n{knowledge_base}\n\nChat history:\n"
+    for user_msg, bot_msg in history:
+        context_text += f"User: {user_msg}\nBot: {bot_msg}\n"
+    context_text += f"User: {user_input}"
+
     # Check if user input exists in knowledge base
     if user_input.lower() in knowledge_base.lower():
         reply = "📖 Based on knowledge base: " + user_input
@@ -32,12 +40,12 @@ def chat(user_input, history):
         try:
             client = genai.Client(api_key=global_api_key)
             chat_session = client.chats.create(model="gemini-2.5-flash")
-            response = chat_session.send_message(user_input)
+            response = chat_session.send_message(context_text, max_output_tokens=300)
             reply = response.text
         except Exception as e:
             reply = f"❌ Error calling Gemini model: {e}"
 
-    history = history + [(user_input, reply)]  # store chat history
+    history.append((user_input, reply))  # store chat history
     return history, ""
 
 # --- Gradio UI ---