Update app.py

app.py

@@ -103,10 +103,11 @@ conversation_chain = ConversationalRetrievalChain.from_llm(
     retriever=vectorstore.as_retriever(),
     memory=memory,
     combine_docs_chain_kwargs={"prompt": qa_prompt},
-    return_source_documents=True,
-    output_key='answer'  # Specify output_key to fix the error
+    return_source_documents=False,  # Changed to False
+    chain_type="stuff"
 )
 
+
 def audio_frame_callback(frame: av.AudioFrame) -> av.AudioFrame:
     audio = frame.to_ndarray().flatten()
     st.session_state.audio_buffer.put(audio)
@@ -122,12 +123,10 @@ def get_ai_response(user_input: str) -> str:
             if len(user_input) > 512:
                 user_input = user_input[:512]
 
-            response = conversation_chain({
-                "question": user_input,
-                "chat_history": memory.chat_memory.messages[-5:]
-            })
-
-            if not response or 'answer' not in response:
+            # Update response handling
+            response = conversation_chain({"question": user_input})
+            
+            if not response:
                 if attempt < max_retries - 1:
                     time.sleep(2 ** attempt)
                     continue