Update app.py

app.py

@@ -2,9 +2,16 @@ import json
 import requests
 import gradio as gr
 
+STYLE = """
+#chatbot {
+    height: 700px !important;
+}
+"""
+
 def download_fn(url):
     resp = requests.get(url)
     data = json.loads(resp.content)
+    print(len(data))
 
     for idx, data_item in enumerate(data[:]):
         for conv in data_item:
@@ -65,7 +72,7 @@ def move_prev_fn(data, cursor):
         gr.Button(interactive=True)
     )
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=STYLE) as demo:
     data = gr.State([])
     data_to_be_removed = gr.State(set())
     cursor = gr.State(-1)
@@ -74,13 +81,17 @@ with gr.Blocks() as demo:
         url = gr.Textbox(label="URL", placeholder="URL to download conversation recorded JSON", scale=9)
         download = gr.Button("Download", scale=1)
     
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(elem_id="chatbot")
 
     with gr.Row():
         prev_btn = gr.Button("Prev", interactive=False)
         mark_btn = gr.Button("✅ Mark to remove ", interactive=False)
         next_btn = gr.Button("Next", interactive=False)
 
+    with gr.Row():
+        cleanup_btn = gr.Button("Clean-up")
+        get_btn = gr.Button("Download clean-up version")
+
     download.click(
         download_fn,
         inputs=[url], 
@@ -107,4 +118,4 @@ with gr.Blocks() as demo:
 
     
 
-demo.launch(debug=True)
+demo.launch()