Update app.py

app.py

@@ -1,33 +1,40 @@
 import gradio as gr
 
 with gr.Blocks() as demo:
-    history = gr.State([])
-    user_question = gr.State("")
-    output_file_name = "chat_history.json"
+    chat_history = gr.State(value=[])
     
-    def slow_echo(message, history):
-        for i in range(len(message)):
-            yield "You typed: " + message[: i+1]
+    def echo(message, history):
+        response = "You typed: " + message
+        chat_history.value.append(
+            {
+                "user": message,
+                "bot": response,
+            }
+        )
+        return response
 
-    def generate_json(history):
-        pass
+    def generate_json(chat_history):
+        return chat_history.value
         
     chatbox = gr.ChatInterface(
-        fn=slow_echo,
-        examples=["How can I help you?"],
+        fn=echo,
+        
         title="Title Here",
         description="Description for the task",
+        examples=["How can I help you?"],
+        
         submit_btn="Enter",
         stop_btn="Stop generating",
         retry_btn="Regenerate",
         undo_btn="Undo last message",
         clear_btn="Start a new conversation"
-    )
-    chatbox.queue()
-
-    json_gen_btn = gr.Button("📩 Download the JSON file for your chat history!")
-    chat_history_json = []
-    json_gen_btn.click(fn=generate_json, inputs=history, outputs=chat_history_json)
-
+    ).queue()
+    
+    chat_history_json = gr.JSON(generate_json(chat_history))
+    gr.Markdown("### 📩 Generate the JSON file for your chat history!")
+    gr.Interface(fn=generate_json, 
+                 inputs=None, 
+                 outputs=[ chat_history_json ])
+    
 demo.queue()
 demo.launch()