Update app.py

app.py

@@ -251,7 +251,7 @@ Answer:"""
 # Create the QA chain with correct variables
 memory = ConversationBufferMemory(
     memory_key="chat_history",
-    input_key="question",
+    input_key="query",
     output_key="answer"
 )
 
@@ -302,7 +302,7 @@ Answer:"""
 
 fallback_prompt = PromptTemplate(
     template=fallback_template,
-    input_variables=["question", "chat_history"]
+    input_variables=["query", "chat_history"]
 )
 fallback_chain = LLMChain(
     llm=llm,
@@ -344,17 +344,13 @@ if submit_button and query:
             st.write("🔍 Retrieving relevant information...")
             st.write("📝 Generating personalized response...")
             chat_history = st.session_state.conversation_memory.load_memory_variables({}).get("chat_history", "")
-            # Generate response
-            result = qa({
-            "query": query,
-            })
-            
-            if result['result'].strip() == "":
-                response = fallback_chain.run(
-                    question=query,
-                    chat_history=st.session_state.conversation_memory.load_memory_variables({})["chat_history"])
-            else:
-                response = result['result']
+            try:
+                result = qa({
+                    "query": query  # Changed from "query" to "question"
+                })
+                response = result['result'] if result['result'].strip() != "" else fallback_chain.run(query=query, chat_history=st.session_state.conversation_memory.load_memory_variables({})["chat_history"])
+            except Exception as e:
+                response = fallback_chain.run(query=query, chat_history=st.session_state.conversation_memory.load_memory_variables({})["chat_history"])
 
             st.session_state.conversation_memory.save_context(
                 {"input": query},
@@ -363,10 +359,8 @@ if submit_button and query:
             
             # Display final response
             st.markdown(response)
-    
-    # Add assistant response to history
-    st.session_state.messages.append({"role": "assistant", "content": response})
-    
+            st.session_state.messages.append({"role": "assistant", "content": response})
+
     # Clear the form
     st.session_state["query"] = ""