Update app.py

app.py

@@ -9,12 +9,9 @@ print("Loading sentence-transformer model for retrieval...")
 retriever_model = SentenceTransformer('all-MiniLM-L6-v2')
 print("Retriever model loaded.")
 
-# --- THIS IS THE UPDATED LINE ---
 print("Loading generative model for answering (google/flan-t5-base)...")
-# Using the balanced 'base' model for better performance and reliability.
 generator_pipe = pipeline("text2text-generation", model="google/flan-t5-base", device=-1)
 print("Generative model loaded.")
-# --- END OF UPDATE ---
 
 
 # --- 2. Setup ChromaDB ---
@@ -71,12 +68,14 @@ except ValueError:
     print("ChromaDB collection loaded.")
 
 
-# --- 3. Define Chatbot Logic ---
+# --- 3. Define Chatbot Logic (UPDATED SECTION) ---
 def chatbot_response(message, history):
     query_embedding = retriever_model.encode([message]).tolist()
+    
+    # --- 1. INCREASED CONTEXT ---
     results = collection.query(
         query_embeddings=query_embedding,
-        n_results=5 # Using 5 results is a good balance for the base model
+        n_results=10 # Retrieve more context
     )
     retrieved_documents = results['documents'][0]
     
@@ -84,8 +83,12 @@ def chatbot_response(message, history):
         return "I'm sorry, I couldn't find any relevant information in the chat history. 🤔"
     
     context = "\n- ".join(retrieved_documents)
+    
+    # --- 2. ENHANCED PROMPT ---
     prompt = f"""
-    Based on the following excerpts from a WhatsApp chat, provide a helpful and accurate answer to the user's question.
+    Based ONLY on the following excerpts from a WhatsApp chat, provide a detailed and comprehensive answer to the user's question. 
+    Synthesize the information from all relevant excerpts into a single, well-structured response.
+    Do not make up information.
 
     Chat Context:
     - {context}
@@ -93,10 +96,11 @@ def chatbot_response(message, history):
     Question:
     {message}
 
-    Answer:
+    Detailed Answer:
     """
     
-    generated_text = generator_pipe(prompt, max_length=150, num_beams=5, early_stopping=True)
+    # --- 3. INCREASED MAX LENGTH ---
+    generated_text = generator_pipe(prompt, max_length=400, num_beams=5, early_stopping=True)
     response = generated_text[0]['generated_text']
     
     return response
@@ -104,10 +108,10 @@ def chatbot_response(message, history):
 # --- 4. Create the Gradio Interface ---
 iface = gr.ChatInterface(
     fn=chatbot_response,
-    title="WhatsApp Chat Bot ⚡️",
-    description="Ask me anything about this WhatsApp chat history. (Powered by flan-t5-base)",
+    title="WhatsApp Chat Bot ✍️",
+    description="Ask me anything about this WhatsApp chat history. I will provide a detailed response.",
     theme="soft", 
-    examples=["What was the final decision on the project deadline?", "Summarize the conversation about the event."],
+    examples=["Summarize the conversation about the event in detail.", "What were all the points raised about the project deadline?"],
     cache_examples=False
 )