Update app.py

app.py

@@ -401,17 +401,20 @@ with gr.Blocks(title="Enhanced JSON Chatbot") as demo:
 
     status = gr.Markdown("🔄 Ready. Upload JSON file to begin.")
     
+    chatbot = gr.Chatbot(height=500)
+    
+    with gr.Row():
+        user_box = gr.Textbox(
+            label="Ask about your JSON data...", 
+            placeholder="e.g., Who are the batch instructors? or Who is Shukdev Datta?",
+            lines=2,
+            scale=4
+        )
+        send_btn = gr.Button("Send", variant="primary", size="lg", scale=1)
+    
     with gr.Row():
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(height=500)
-            user_box = gr.Textbox(
-                label="Ask about your JSON data...", 
-                placeholder="e.g., Who are the batch instructors? or Who is Shukdev Datta?",
-                lines=2
-            )
-        with gr.Column(scale=1):
-            clear_btn = gr.Button("Clear Chat", variant="secondary", size="sm")
-            reload_btn = gr.Button("Reload JSON", variant="secondary", size="sm")
+        clear_btn = gr.Button("Clear Chat", variant="secondary")
+        reload_btn = gr.Button("Reload JSON", variant="secondary")
 
     # States
     state_corpus = gr.State([])
@@ -491,6 +494,12 @@ with gr.Blocks(title="Enhanced JSON Chatbot") as demo:
         outputs=[chatbot],
     ).then(lambda: "", outputs=[user_box])  # Clear input after submit
 
+    send_btn.click(
+        ask_enhanced,
+        inputs=[api_key_tb, user_box, chatbot, state_corpus, topk_slider, per_table_cap],
+        outputs=[chatbot],
+    ).then(lambda: "", outputs=[user_box])  # Clear input after send
+
     reload_btn.click(
         load_json_to_corpus,
         inputs=[json_file, fallback_path, max_val_len],