Update app.py

app.py

@@ -155,7 +155,7 @@ prompt = ChatPromptTemplate.from_messages([
     ("human", "Context:\n{context}\n\nQuestion: {question}")
 ])
 
-llm = ChatOpenAI(model="gpt-4o", openai_api_key=OPENAI_API_KEY, temperature=0)
+llm = ChatOpenAI(model="gpt-4.1", openai_api_key=OPENAI_API_KEY, temperature=0)
 
 retriever = SQLiteVectorRetriever(top_k=5)
 qa_chain = RetrievalQA.from_chain_type(
@@ -204,8 +204,8 @@ def send_message():
         return
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.spinner("Thinking..."):
-        # Use the chain with {"question": ...} to match prompt format
-        result = qa_chain({"question": user_input})
+        # Correct input key: "query" (not "question")
+        result = qa_chain({"query": user_input})
         answer = result['result']
         st.session_state.messages.append({"role": "assistant", "content": answer})
         docs = result['source_documents']
@@ -219,6 +219,7 @@ def send_message():
         st.session_state.messages.append({"role": "function", "content": json.dumps(doc_list, indent=2)})
     st.session_state.temp_input = ""
 
+
 st.text_input("Your message:", key="temp_input", on_change=send_message)
 
 if st.button("Clear chat"):