app.py chat

app.py

@@ -156,19 +156,19 @@ def load_db(file, k):
 
 chat_history = []  # initialize chat history
 
-def chatbot_conversation(interim_responses, pdf_file):
+def greet(messages):
     global chat_history
-    question = interim_responses[-1]["message"] if interim_responses else "Hello"
-    a = load_db(pdf_file, 3)
-    r = a.invoke({"question": question, "chat_history": chat_history})
+    user_message = messages[-1]['content']
+    a = load_db("temp.pdf", 3)  # assuming you've uploaded the file and saved it as "temp.pdf"
+    r = a.invoke({"question": user_message, "chat_history": chat_history})
     match = re.search(r'Helpful Answer:(.*)', r['answer'])
     if match:
         helpful_answer = match.group(1).strip()
         # Extend chat history with the current question and answer
-        chat_history.extend([(question, helpful_answer)])
-        return helpful_answer, None
+        chat_history.extend([{"role": "user", "content": user_message}, {"role": "assistant", "content": helpful_answer}])
+        return [{"role": "assistant", "content": helpful_answer}]
     else:
-        return "No helpful answer found.", None
+        return [{"role": "assistant", "content": "No helpful answer found."}]
 
-iface = gr.ChatInterface(fn=chatbot_conversation, inputs=["text", "file"], outputs="text")
+iface = gr.ChatInterface(fn=greet, inputs=gr.Textbox(), outputs=gr.ChatResponse(continuous_update=True))
 iface.launch(share=True)