Update app.py

app.py

@@ -91,14 +91,11 @@ def generate_chunked_response(model, prompt, max_tokens=500, max_chunks=5):
 def response(database, model, question):
     prompt_val = ChatPromptTemplate.from_template(prompt)
     retriever = database.as_retriever()
-    
     context = retriever.get_relevant_documents(question)
     context_str = "\n".join([doc.page_content for doc in context])
-    
     formatted_prompt = prompt_val.format(context=context_str, question=question)
-    
     ans = generate_chunked_response(model, formatted_prompt)
-    return ans  # Only return the final answer
+    return ans.split("Question:")[-1].strip()  # Return only the answer part
 
 def update_vectors(files, use_recursive_splitter):
     if not files: