Update app.py

app.py

@@ -170,29 +170,30 @@ prompt = ChatPromptTemplate.from_messages(
 chain_neo4j = (
     RunnableParallel(
         {
-            "chat_history": _search_query,
-            "input": RunnablePassthrough(),
-            "agent_scratchpad": retriever_neo4j,
+            "chat_history": _search_query,  # Pass chat history through the _search_query
+            "input": RunnablePassthrough(),  # Directly pass user input as 'input'
+            "agent_scratchpad": retriever_neo4j,  # Use the retriever to get additional context
         }
     )
-    | prompt
-    | llm
-    | StrOutputParser()
+    | prompt  # Use the prompt with the provided context
+    | llm  # Use the LLM for generating a response
+    | StrOutputParser()  # Parse the response into a string
 )
 
 # Define the function to get the response
 def get_response(input_text):
     try:
-        # Adjust input to match expected structure
+        # Call the chain with the proper variables
         return chain_neo4j.invoke({
-            "input": input_text,
+            "input": input_text,  # User's question as 'input'
             "chat_history": [],  # Replace with actual chat history if available
-            "agent_scratchpad": ""  # Replace with actual content if needed
+            "agent_scratchpad": ""  # Placeholder for any additional data
         })
     except Exception as e:
         return f"Error: {str(e)}"
 
 
+
 # Define the function to clear input and output
 def clear_fields():
     return [],"",None