Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,43 +1,53 @@
-import os
-import re
-import shutil
-import zipfile
-import asyncio
-import tempfile
-from core_creator.app_blueprint import build_app_code
-from core_creator.voice_to_app import robot_action
-
-async def deploy_callback(idea: str):
+# gradio_generator.py - Gradio interface for voice-to-robot app generation
+
+import gradio as gr
+from core_creator.voice_to_app import VoiceToAppCreator
+
+def deploy_callback(voice_input: str) -> dict:
     """
-    Generate the app directory, package into zip, return status and path.
+    Processes the user's voice input to generate a robot app package.
     """
-    # Normalize name
-    safe_name = re.sub(r"[^a-z0-9]+", "_", idea.lower()).strip("_") or "app"
-    with tempfile.TemporaryDirectory() as tmpdir:
-        app_dir = os.path.join(tmpdir, f"robosage_{safe_name}")
-        try:
-            # Build code files
-            os.makedirs(app_dir, exist_ok=False)
-            build_app_code(idea, app_dir)
-        except Exception as e:
-            return f"❌ Failed to generate app directory for: {idea}", None
-
-        # Create zip
-        zip_path = os.path.join(tmpdir, f"robosage_{safe_name}.zip")
-        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-            for root, _, files in os.walk(app_dir):
-                for file in files:
-                    full = os.path.join(root, file)
-                    arc = os.path.relpath(full, app_dir)
-                    zf.write(full, arc)
-
-        return f"✅ Generated app: {idea}", zip_path
-
-def robot_behavior(command: str):
+    creator = VoiceToAppCreator(voice_input)
+    app_package = creator.run_pipeline()
+    return app_package
+
+def robot_behavior(voice_input: str) -> str:
     """
-    Simulate robot behavior: map command to action.
+    Generates a summary of the robot's behavior based on the voice input.
     """
-    try:
-        return robot_action(command)
-    except Exception:
-        return f"❓ Unknown action: {command}"
+    creator = VoiceToAppCreator(voice_input)
+    app_package = creator.run_pipeline()
+    return app_package["blueprint"].get("description", "No description available.")
+
+# Gradio Interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 🤖 Voice-to-Robot App Generator")
+    gr.Markdown("Describe your robot idea, and we'll generate an app blueprint for you.")
+
+    with gr.Row():
+        voice_input = gr.Textbox(label="🗣️ Your Robot Idea", placeholder="e.g., A robot that helps with home cleaning tasks.")
+        submit_btn = gr.Button("Generate App")
+
+    with gr.Row():
+        intent_output = gr.Textbox(label="🔍 Detected Intent")
+        description_output = gr.Textbox(label="📄 App Description")
+        code_output = gr.Code(label="💻 Generated Code", language="python")
+        assets_output = gr.JSON(label="🎨 Visual/Audio Assets")
+
+    def process_input(user_input):
+        app_package = deploy_callback(user_input)
+        return (
+            app_package.get("intent", ""),
+            app_package["blueprint"].get("description", ""),
+            app_package.get("code", ""),
+            app_package.get("assets", {})
+        )
+
+    submit_btn.click(
+        fn=process_input,
+        inputs=voice_input,
+        outputs=[intent_output, description_output, code_output, assets_output]
+    )
+
+if __name__ == "__main__":
+    demo.launch()