Update app.py

app.py

@@ -139,13 +139,26 @@ Answer concisely:"""
 # Create a prompt object using the template
 prompt = ChatPromptTemplate.from_template(template)
 
+# Function to generate a response using the prompt and the context
+#def generate_response_with_prompt(context, question):
+    #response = prompt.format(
+        #context=context,
+        #question=question
+    #)
+    #return response
+
 # Function to generate a response using the prompt and the context
 def generate_response_with_prompt(context, question):
-    response = prompt.format(
+    formatted_prompt = prompt.format(
         context=context,
         question=question
     )
-    return response
+    # Use the ChatOpenAI instance to generate a response directly from the formatted prompt
+    llm = ChatOpenAI(temperature=0, api_key=os.environ['OPENAI_API_KEY'])
+    response = llm(formatted_prompt)
+    return response.content.strip()
+
+
 
 # Define the function to generate a hybrid response using Neo4j and other retrieval methods
 def retriever(question: str):