Update app.py

app.py

@@ -166,10 +166,6 @@ prompt = ChatPromptTemplate.from_messages(
     ]
 )
 
-# Ensure 'context' and 'question' map correctly to the prompt's expected variables
-qa_prompt = prompt
-
-
 # Define the chain for Neo4j-based retrieval and response generation
 chain_neo4j = (
     RunnableParallel(
@@ -179,16 +175,20 @@ chain_neo4j = (
             "agent_scratchpad": retriever_neo4j,
         }
     )
-    | qa_prompt
+    | prompt
     | llm
     | StrOutputParser()
 )
 
-
 # Define the function to get the response
-def get_response(question):
+def get_response(input_text):
     try:
-        return chain_neo4j.invoke({"input": question, "chat_history": [], "agent_scratchpad": ""})
+        # Adjust input to match expected structure
+        return chain_neo4j.invoke({
+            "input": input_text,
+            "chat_history": [],
+            "agent_scratchpad": ""
+        })
     except Exception as e:
         return f"Error: {str(e)}"