Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,9 +1,9 @@
-# gradio_generator.py - Wraps robot behavior and simulator into Gradio interface
+# gradio_generator.py - Safe version for Hugging Face Spaces
 
 import gradio as gr
 from deployer.simulator_interface import VirtualRobot
 
-# The logic that interprets voice or text and maps it to robot actions
+# Logic that interprets voice or text and maps it to robot actions
 def robot_behavior(user_input: str):
     bot = VirtualRobot()
     user_input = user_input.lower().strip()
@@ -15,22 +15,20 @@ def robot_behavior(user_input: str):
     else:
         return bot.perform_action("say I didn't understand that. Try again!")
 
-# Generate Gradio app dynamically
+# Generate Gradio app
 def launch_gradio_app(title="RoboSage App", description="Your robot, your voice."):
-    with gr.Blocks() as demo:
-        gr.Markdown(f"""# 🤖 {title}
-{description}
-        """)
-        with gr.Row():
+    def create_ui():
+        with gr.Blocks() as demo:
+            gr.Markdown(f"# 🤖 {title}\n{description}")
             inp = gr.Textbox(label="Speak or Type")
             out = gr.Textbox(label="Robot Response")
-        btn = gr.Button("Send")
+            btn = gr.Button("Send")
+            btn.click(fn=robot_behavior, inputs=inp, outputs=out)
+        return demo
 
-        btn.click(fn=robot_behavior, inputs=inp, outputs=out)
+    return create_ui()
 
-    return demo
-
-# For testing
+# Run directly
 if __name__ == "__main__":
     app = launch_gradio_app()
     app.launch()