feat: Expose Gradio Blocks as demo for Hugging Face Spaces compatibility

Gradio_UI.py

@@ -292,5 +292,35 @@ class GradioUI:
 
         demo.launch(debug=True, **kwargs)
 
+    def build_blocks(self):
+        import gradio as gr
+
+        with gr.Blocks(fill_height=True) as demo:
+            stored_messages = gr.State([])
+            file_uploads_log = gr.State([])
+            chatbot = gr.Chatbot(
+                label="Agent",
+                type="messages",
+                avatar_images=(
+                    None,
+                    "https://huggingface.co/datasets/agents-course/course-images/resolve/main/en/communication/Alfred.png",
+                ),
+                resizeable=True,
+                scale=1,
+            )
+            # If an upload folder is provided, enable the upload feature
+            # (Assume file upload is not required for Spaces demo, can be added if needed)
+            text_input = gr.Textbox(lines=1, label="Chat Message")
+            # Minimal interaction: just append user message
+            def interact_with_agent(prompt, messages):
+                messages.append(gr.ChatMessage(role="user", content=prompt))
+                return messages
+            text_input.submit(
+                interact_with_agent,
+                [text_input, stored_messages],
+                [stored_messages],
+            )
+        return demo
+
 
 __all__ = ["stream_to_gradio", "GradioUI"]

app.py

@@ -266,5 +266,5 @@ agent = CustomCodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Configure Gradio UI with sharing enabled
-GradioUI(agent).launch(ssr_mode=False)
+# Expose the Gradio Blocks app as 'demo' for Hugging Face Spaces
+demo = GradioUI(agent).build_blocks()