refactor: Enhance chat handling in Gradio application by implementing textbox interactivity during streaming phases, updating UI elements for improved user experience, and refining the application title for clarity.

app.py

@@ -150,7 +150,7 @@ def should_use_tools(user_msg: str) -> bool:
 
 
 # -------------------------
-# Main chat handler (streaming)
+# Main chat handler (streaming + disabling textbox)
 # -------------------------
 
 
@@ -165,6 +165,7 @@ def chat_send_stream(user_msg, history, file_url):
         Phase 2: streaming final answer via Responses API (no tools)
     - Keeps full chat history so follow-ups work.
     - Shows status/progress messages in the UI when tools are used.
+    - Disables the textbox during work, re-enables at the end.
     """
 
     # UI history (what Gradio displays)
@@ -186,7 +187,11 @@ def chat_send_stream(user_msg, history, file_url):
     if not use_tools:
         # Add a small status bubble then stream
         history.append({"role": "assistant", "content": "✏️ Generating answer..."})
-        yield history
+        # Disable textbox while generating
+        yield (
+            history,
+            gr.update(interactive=False),
+        )
 
         # Build input text for Responses API
         input_text = (
@@ -208,10 +213,18 @@ def chat_send_stream(user_msg, history, file_url):
             if event.type == "response.output_text.delta":
                 final_text += event.delta
                 history[-1]["content"] = final_text
-                yield history
+                yield (
+                    history,
+                    gr.update(interactive=False),
+                )
             elif event.type == "response.completed":
                 break
 
+        # Re-enable textbox at the end
+        yield (
+            history,
+            gr.update(interactive=True, value=""),
+        )
         return
 
     # -------------------------
@@ -229,7 +242,11 @@ def chat_send_stream(user_msg, history, file_url):
                 ),
             }
         )
-        yield history
+        # Keep textbox enabled because nothing heavy is happening
+        yield (
+            history,
+            gr.update(interactive=True),
+        )
         return
 
     # User message for the model includes the CSV URL
@@ -246,7 +263,11 @@ def chat_send_stream(user_msg, history, file_url):
             "content": "⏳ Analyzing your request and selecting MCP tools...",
         }
     )
-    yield history
+    # Disable textbox while tools run
+    yield (
+        history,
+        gr.update(interactive=False),
+    )
 
     # Build a single string input for the tool phase
     tool_phase_input = (
@@ -274,7 +295,11 @@ def chat_send_stream(user_msg, history, file_url):
         "role": "assistant",
         "content": "✅ MCP tools finished. Preparing explanation...",
     }
-    yield history
+    # Keep textbox disabled (we're about to stream final answer)
+    yield (
+        history,
+        gr.update(interactive=False),
+    )
 
     # -------------------------
     # Phase 2: Final streaming explanation
@@ -306,24 +331,32 @@ def chat_send_stream(user_msg, history, file_url):
         if event.type == "response.output_text.delta":
             final_text += event.delta
             history[-1]["content"] = final_text
-            yield history
+            yield (
+                history,
+                gr.update(interactive=False),
+            )
         elif event.type == "response.completed":
             break
 
+    # Re-enable textbox at the end, and clear it
+    yield (
+        history,
+        gr.update(interactive=True, value=""),
+    )
+
 
 # -------------------------
 # Gradio UI
 # -------------------------
 
-with gr.Blocks(title="MCP + GPT-5 mini - Streaming MLOps Agent") as demo:
+with gr.Blocks(title="Streaming MLOps Agent") as demo:
     gr.Markdown(
         """
-        # AI-Driven MLOps Agent 🤖
-
-        - You can just chat (e.g. "hey") — no tools needed.
-        - For data work, **upload a CSV file** and ask to analyze, train, or deploy.
-        - When tools are used, you’ll see status updates.
-        - Final answers stream token by token.
+        # 🧠 Smart MLOps Agent
+        
+        - 💬 Chat naturally, even just “hey”
+        - 📂 Upload CSVs for analysis, training, and deployment
+        - ⚡ See live tool status and streaming answers
         """
     )
 
@@ -355,24 +388,19 @@ with gr.Blocks(title="MCP + GPT-5 mini - Streaming MLOps Agent") as demo:
         interactive=True,
         placeholder="Say hi, or ask me to analyze / train / deploy on your dataset...",
     )
-    send = gr.Button("Send", interactive=True)
-
-    send.click(
-        chat_send_stream,
-        inputs=[msg, chatbot, file_url_state],
-        outputs=[chatbot],
-    ).then(lambda: "", outputs=[msg])
 
+    # Only Enter/Return sends messages; no Send button
     msg.submit(
         chat_send_stream,
         inputs=[msg, chatbot, file_url_state],
-        outputs=[chatbot],
-    ).then(lambda: "", outputs=[msg])
-
+        outputs=[chatbot, msg],
+    )
 
 if __name__ == "__main__":
     demo.queue().launch(
+        theme=gr.themes.Soft(),
         allowed_paths=["/tmp"],
         ssr_mode=False,
         show_error=True,
-        )
+        quiet=True,
+    )