Update app.py

app.py

@@ -77,9 +77,16 @@ def generate_response(user_message):
             full_response += new_token
     thread.join()
     return full_response
+
 # -----------------------------
 # 5️⃣ Chat + Save
 # -----------------------------
+chat_history = []
+
+def chat(user_message):
+    bot_response = generate_response(user_message)
+    chat_history.append((user_message, bot_response))
+    return chat_history, ""  # also clears input box
 
 def save_conversation():
     if not chat_history:
@@ -91,7 +98,7 @@ def save_conversation():
         conversation.append({"role": "assistant", "content": str(bot_msg)})
     
     timestamp = time.strftime("%Y%m%d-%H%M%S")
-    file_path = f"conversation_{timestamp}.txt"
+    file_path = f"conversation_{timestamp}.txt"   # save as TXT not JSON
     
     with open(file_path, "w", encoding="utf-8") as f:
         json.dump(conversation, f, indent=4, ensure_ascii=False)
@@ -99,22 +106,22 @@ def save_conversation():
     return file_path
 
 # -----------------------------
-# 5️⃣ Gradio interface
+# 6️⃣ Gradio interface
 # -----------------------------
-
 with gr.Blocks() as demo:
     gr.Markdown("# Nigerian PIDGIN Assistant")
     gr.Markdown("Chat with a Nigerian assistant that only speaks Pidgin English.")
 
-    chatbot = gr.Textbox(label="Conversation", lines=15, interactive=False)
+    chatbot = gr.Chatbot(label="Conversation")
     user_input = gr.Textbox(label="Your message", placeholder="Type your message here...")
     
     with gr.Row():
         send_button = gr.Button("Send")
         save_button = gr.Button("Save Conversation")
+        download_file = gr.File(label="Download Conversation")
 
-    send_button.click(generate_response, inputs=user_input, outputs=chatbot)
-    save_button.click(save_conversation, outputs=gr.File())
+    send_button.click(chat, inputs=user_input, outputs=[chatbot, user_input])
+    save_button.click(save_conversation, outputs=download_file)
 
 demo.launch()
 # iface = gr.Interface(