Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,42 +1,27 @@
-# gradio_generator.py - Gradio UI for Voice-to-App Creator
+# gradio_generator.py - Launches the Gradio UI for robot app deployment
 
 import gradio as gr
 from core_creator.voice_to_app import VoiceToAppCreator
 
-def deploy_callback(voice_input):
-    """
-    Callback function to process voice input and generate the robot app package.
-    """
-    creator = VoiceToAppCreator(voice_input)
-    app_package = creator.run_pipeline()
-    return app_package
+def deploy_callback(user_input: str) -> dict:
+    creator = VoiceToAppCreator(user_input)
+    return creator.run_pipeline()
 
-def robot_behavior():
-    """
-    Defines the Gradio interface for the Voice-to-App Creator.
-    """
-    with gr.Blocks() as demo:
-        gr.Markdown("# 🤖 Voice-to-App Creator")
-        gr.Markdown("Describe your robot idea, and we'll generate an app for it!")
+def robot_behavior(user_input: str) -> str:
+    result = deploy_callback(user_input)
+    return result.get("code", "No code generated.")
 
+def launch_gradio_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🤖 Voice-to-Robot App Generator")
         with gr.Row():
-            voice_input = gr.Textbox(
-                label="Your Robot Idea",
-                placeholder="e.g., Build a robot that teaches kids to brush their teeth with fun animations.",
-                lines=3
-            )
-            submit_button = gr.Button("Generate App")
-
-        output = gr.JSON(label="Generated App Package")
+            input_box = gr.Textbox(label="Describe your robot idea", placeholder="e.g., A robot that helps kids brush their teeth.")
+            output_box = gr.Code(label="Generated Code", language="python")
+        submit_btn = gr.Button("Generate App")
 
-        submit_button.click(
-            fn=deploy_callback,
-            inputs=voice_input,
-            outputs=output
-        )
+        submit_btn.click(fn=robot_behavior, inputs=input_box, outputs=output_box)
 
-    return demo
+    demo.launch()
 
 if __name__ == "__main__":
-    demo = robot_behavior()
-    demo.launch()
+    launch_gradio_interface()