Update app.py

app.py

@@ -99,7 +99,11 @@ def chat(message,history):
             similar_chat_history += chats
 
     #prompt = f"<|user|>\n{message}<|end|>\n<|assistant|>"
-    user_input = similar_chat_history + formatted_user_msg
+
+    top_7_cells_string = '\n'.join(top_7_cells)
+    context_plus_message = top_7_cells_string  + message['text']
+    formatted_context_plus_message = user_prompt_template(context_plus_message)
+    user_input = similar_chat_history + formatted_context_plus_message
     
     # print(user_input)
     # print('-'*20)
@@ -120,8 +124,7 @@ def chat(message,history):
 
     
         
-    # return formatted_assistant_msg
-    return '\n'.join(top_7_cells)
+    return formatted_assistant_msg
 
 demo = gr.ChatInterface(chat, multimodal=True)