Update app.py

app.py

@@ -46,7 +46,7 @@ Recommendation:
 
 PROMPT = PromptTemplate(
     template=prompt_template,
-    input_variables=["chat_history", "query", "context"]
+    input_variables=["chat_history", "question", "context"]
 )
 
 # Initialize the language model
@@ -86,8 +86,8 @@ if prompt := st.chat_input("What are you looking to learn?"):
 
     # Assistant response generation with streaming effect
     with st.chat_message("assistant"):
-        response = qa_chain({"query": prompt})
-        response_text = response["response"]
+        response = qa_chain({"question": prompt})
+        response_text = response["answer"]
 
         # Stream the response word by word
         for accumulated_response in response_text: