Update app.py

app.py

@@ -542,6 +542,29 @@ def answer_formatter_node(question, context):
     res = response.content
     return res
 
+def reguar_answer_node(question, context):
+    prompt = f""" You are an highly clever question-answering assistant trained to provide clear and concise answers based on the user query and provided context. 
+    Your task is to generated answers for the user query based on the context provided.
+    Instructions for your response:
+    1. Directly answer the user query. Make use of provided context if necessary.
+    2. Ensure your response is clear and concise.
+    3. Give the answer in json format with a single key named 'content' with value as your response.
+    4. It is important to give response in json format, otherwise the chat may terminate.
+
+    Please format an answer based on the following user question and context provided:
+
+    User Question: 
+    {question}
+
+    Context:
+    {context}
+    """
+    response = answer_formatter.invoke(
+        [SystemMessage(content=prompt)]
+    )
+    res = response.content
+    return res
+
 CURR_CONTEXT = ''
 
 # @spaces.GPU
@@ -666,11 +689,17 @@ def handle_message(data):
                 'input': message,
                 'context': json.dumps(CURR_CONTEXT)
             }
-            response = json_answer_generator(message, context)
+            response = answer_generator(formated_input, session_id=session_id)
             emit('response', response, room=session_id)
-        else:
+        elif task == "recommendation":
             response = recommendation_node(message)
             # response = answer_formatter_node(message, recipes)
+            if response is None:
+                response = {'content':"An error occurred while processing your request."}
+
+            emit('json_response', response, room=session_id)
+        else:
+            response = reguar_answer_node(message, CURR_CONTEXT)
             if response is None:
                 response = {'content':"An error occurred while processing your request."}
             emit('json_response', response, room=session_id)