Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,77 +1,97 @@
 import os
-import asyncio
 import gradio as gr
-
+import openai
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
-from deployer.revenue_tracker import package_artifacts
-
-# --- Async pipeline runner ---
-async def _run_pipeline_async(idea: str) -> str:
-    creator = VoiceToAppCreator(idea)
-    assets = creator.run_pipeline()
-    # blueprint title fallback
-    return assets.get("blueprint", {}).get("title", "<unknown>")
-
-# --- Sync wrappers for Gradio ---
-def generate_app(idea: str) -> str:
-    title = asyncio.run(_run_pipeline_async(idea))
-    return f"✅ Generated app: {title}"
 
-def get_app_zip() -> str:
-    zip_path = package_artifacts()
-    if not os.path.isfile(zip_path):
-        raise FileNotFoundError(f"Could not find ZIP at {zip_path}")
-    return zip_path
+# Ensure API key is set
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OPENAI_API_KEY:
+    raise EnvironmentError(
+        "Please set the OPENAI_API_KEY environment variable for audio transcription."
+    )
+openai.api_key = OPENAI_API_KEY
 
-def robot_behavior(command: str) -> str:
-    return VirtualRobot().perform_action(command)
 
-# --- Build & launch Gradio UI ---
-def launch_gradio_app():
-    css = r"""
-    .gradio-container { max-width: 800px; margin: auto; }
-    .gradio-box { padding: 1em; margin-bottom: 2em; border: 1px solid #e2e2e2; border-radius: 8px; }
+def deploy_app(text_input: str) -> str:
     """
+    Generate the app package based on text input.
+    """
+    assets = VoiceToAppCreator(text_input).run_pipeline()
+    title = assets.get("blueprint", {}).get("title", "<unknown>")
+    return f"✅ Generated app: {title}"
 
-    with gr.Blocks(css=css) as demo:
-        gr.Markdown("# 🚀 RoboSage")
-        gr.Markdown("Generate your custom robot app and test it live.")
-
-        # Generate App section
-        with gr.Box(elem_id="generate-app", css_class="gradio-box"):
-            gr.Markdown("## 1️⃣ Generate App")
-            idea_input = gr.Textbox(label="Your Robot Idea", placeholder="e.g. A friendly greeting robot", lines=1)
-            gen_btn     = gr.Button("Generate App")
-            status_out  = gr.Textbox(label="Status", interactive=False)
-
-            gen_btn.click(fn=generate_app, inputs=[idea_input], outputs=[status_out])
 
-        # Download ZIP section
-        with gr.Box(elem_id="download-app", css_class="gradio-box"):
-            gr.Markdown("## 📦 Download Packaged App")
-            dl_btn  = gr.Button("Download App ZIP")
-            dl_file = gr.File(label="App ZIP File")
+def transcribe_audio(audio_path: str) -> str:
+    """
+    Transcribe audio file to text using OpenAI Whisper.
+    """
+    if not audio_path:
+        return ""
+    response = openai.Audio.transcribe(
+        model="whisper-1",
+        file=audio_path
+    )
+    return response.get("text", "")
 
-            dl_btn.click(fn=get_app_zip, inputs=None, outputs=[dl_file])
 
-        # Simulator section
-        with gr.Box(elem_id="robot-sim", css_class="gradio-box"):
-            gr.Markdown("## 2️⃣ Robot Simulator")
-            cmd_input  = gr.Textbox(label="Command", placeholder="hello or say You rock!", lines=1)
-            sim_btn    = gr.Button("Send")
-            sim_output = gr.Textbox(label="Robot Response", lines=4, interactive=False)
+def robot_behavior(user_input: str) -> str:
+    """
+    Simulate robot action based on user command.
+    """
+    return VirtualRobot().perform_action(user_input)
 
-            sim_btn.click(fn=robot_behavior, inputs=[cmd_input], outputs=[sim_output])
 
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=int(os.environ.get("PORT", 7860)),
-            share=False
-        )
+def launch_gradio_app():
+    """
+    Builds and returns the Gradio Blocks UI for the RoboSage app.
+    """
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
+
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("## 1️⃣ Generate App")
+                text_idea = gr.Textbox(
+                    label="Your Robot Idea",
+                    placeholder="e.g. A friendly greeting robot."
+                )
+                audio_idea = gr.Audio(
+                    source="microphone",
+                    type="filepath",
+                    label="Or speak your robot idea"
+                )
+                gen_btn = gr.Button("Generate App")
+                status = gr.Textbox(label="Status")
+
+                # Combine transcription or text input
+                def on_generate(text, audio):
+                    transcript = transcribe_audio(audio)
+                    final_input = transcript if transcript else text
+                    return deploy_app(final_input)
+
+                gen_btn.click(
+                    fn=on_generate,
+                    inputs=[text_idea, audio_idea],
+                    outputs=status
+                )
+
+            with gr.Column():
+                gr.Markdown("## 2️⃣ Robot Simulator")
+                cmd_input = gr.Textbox(
+                    label="Command",
+                    placeholder="hello or say You rock!"
+                )
+                sim_btn = gr.Button("Send")
+                resp = gr.Textbox(label="Robot Response", lines=4)
+                sim_btn.click(
+                    fn=robot_behavior,
+                    inputs=cmd_input,
+                    outputs=resp
+                )
 
     return demo
 
-# Expose for external import
-robot_behavior     = robot_behavior
-launch_gradio_app  = launch_gradio_app
+
+if __name__ == "__main__":
+    launch_gradio_app().launch()