Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,97 +1,97 @@
-import os
+# deployer/gradio_generator.py
+
+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
 
-# 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 deploy_app(text_input: str) -> str:
+async def _run_pipeline(idea: str) -> str:
     """
-    Generate the app package based on text input.
+    Asynchronously run the voice→app pipeline and return a status message.
     """
-    assets = VoiceToAppCreator(text_input).run_pipeline()
+    creator = VoiceToAppCreator(idea)
+    assets = creator.run_pipeline()
     title = assets.get("blueprint", {}).get("title", "<unknown>")
     return f"✅ Generated app: {title}"
 
-
-def transcribe_audio(audio_path: str) -> str:
+def deploy_callback(idea: str) -> str:
     """
-    Transcribe audio file to text using OpenAI Whisper.
+    Synchronous wrapper for Gradio: generate the app.
     """
-    if not audio_path:
-        return ""
-    response = openai.Audio.transcribe(
-        model="whisper-1",
-        file=audio_path
-    )
-    return response.get("text", "")
-
+    return asyncio.run(_run_pipeline(idea))
 
 def robot_behavior(user_input: str) -> str:
     """
-    Simulate robot action based on user command.
+    Simulate robot action based on a user command.
     """
     return VirtualRobot().perform_action(user_input)
 
+def download_callback() -> str:
+    """
+    Package all generated app artifacts into a ZIP and return its file path.
+    """
+    zip_path = package_artifacts()
+    return zip_path
 
-def launch_gradio_app():
+def launch_gradio_app() -> gr.Blocks:
     """
-    Builds and returns the Gradio Blocks UI for the RoboSage app.
+    Build and return the Gradio interface.
     """
-    with gr.Blocks() as demo:
-        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
+    demo = gr.Blocks()
+    with demo:
+        gr.Markdown(
+            "## 🚀 RoboSage  \n"
+            "Generate your custom robot app and test it live."
+        )
 
         with gr.Row():
-            with gr.Column():
-                gr.Markdown("## 1️⃣ Generate App")
-                text_idea = gr.Textbox(
+            # ————————————————————————————
+            # 1️⃣ Generate App + Download ZIP
+            # ————————————————————————————
+            with gr.Column(scale=1):
+                gr.Markdown("### 1️⃣ Generate App")
+                idea_input = 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)
+                status_out = gr.Textbox(label="Status", interactive=False)
 
                 gen_btn.click(
-                    fn=on_generate,
-                    inputs=[text_idea, audio_idea],
-                    outputs=status
+                    fn=deploy_callback,
+                    inputs=[idea_input],
+                    outputs=[status_out]
                 )
 
-            with gr.Column():
-                gr.Markdown("## 2️⃣ Robot Simulator")
+                download_btn = gr.Button("📦 Download App ZIP")
+                zip_out = gr.File(label="Your App ZIP")
+
+                download_btn.click(
+                    fn=download_callback,
+                    outputs=[zip_out]
+                )
+
+            # ————————————————————————————
+            # 2️⃣ Robot Simulator
+            # ————————————————————————————
+            with gr.Column(scale=1):
+                gr.Markdown("### 2️⃣ Robot Simulator")
                 cmd_input = gr.Textbox(
-                    label="Command",
+                    label="Speak or Type Command",
                     placeholder="hello or say You rock!"
                 )
-                sim_btn = gr.Button("Send")
-                resp = gr.Textbox(label="Robot Response", lines=4)
-                sim_btn.click(
+                send_btn = gr.Button("Send")
+                resp_out = gr.Textbox(label="Robot Response", lines=4, interactive=False)
+
+                send_btn.click(
                     fn=robot_behavior,
-                    inputs=cmd_input,
-                    outputs=resp
+                    inputs=[cmd_input],
+                    outputs=[resp_out]
                 )
 
     return demo
 
-
 if __name__ == "__main__":
     launch_gradio_app().launch()