Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,117 +1,90 @@
-# gradio_generator.py - CPU & Spaces-compatible Gradio interface with HF Inference API for STT & Monetization
-
+import asyncio
 import os
-import json
-import requests
+import shutil
 import gradio as gr
-from deployer.simulator_interface import VirtualRobot
-from deployer.revenue_tracker import get_revenue_stats, package_artifacts
+
 from core_creator.voice_to_app import VoiceToAppCreator
+from deployer.simulator_interface import VirtualRobot
+from deployer.revenue_tracker import package_artifacts
 
-# Hugging Face Inference API settings
-HF_STT_URL = "https://api-inference.huggingface.co/models/openai/whisper-small"
-HK_API_KEY = os.getenv("HK_API_KEY")
-if not HK_API_KEY:
-    raise EnvironmentError(
-        "Please set the HK_API_KEY environment variable for HuggingFace Inference API.")
-HEADERS = {"Authorization": f"Bearer {HK_API_KEY}"}
-
-# Core robot logic
-def robot_behavior(user_input: str) -> str:
-    bot = VirtualRobot()
-    text = user_input.strip().lower()
-    if any(greet in text for greet in ["hello", "hi", "hey", "welcome"]):
-        return bot.perform_action("wave") + "\n" + bot.perform_action("say Hello there!")
-    if text.startswith("say "):
-        return bot.perform_action(text)
-    return bot.perform_action("say I'm sorry, I didn't understand that.")
-
-# Transcribe via Hugging Face Inference API
-def transcribe_audio(audio_file: str) -> str:
-    with open(audio_file, "rb") as f:
-        data = f.read()
-    response = requests.post(HF_STT_URL, headers=HEADERS, data=data)
-    if response.status_code != 200:
-        return f"❗ Transcription error: {response.status_code}"
-    result = response.json()
-    return result.get("text", "")
-
-# Combined flow
-def transcribe_and_respond(audio_file: str) -> str:
-    text = transcribe_audio(audio_file)
-    return robot_behavior(text)
-
-# Package generation pipeline
-def on_generate(i: str):
-    creator = VoiceToAppCreator(i)
+# Async pipeline runner for generating the robot app
+async def run_pipeline_async(idea: str) -> str:
+    creator = VoiceToAppCreator(idea)
     assets = creator.run_pipeline()
-    zip_path = package_artifacts(assets)
-    blueprint = assets.get("blueprint", {})
-    code_files = assets.get("code", {})
-    code_preview = "\n\n".join([f"# {fname}\n{content}" for fname, content in code_files.items()])
-    return (
-        "✅ App generated successfully!",
-        zip_path,
-        blueprint,
-        code_preview
-    )
-
-# Build full-featured Gradio app
-def launch_gradio_app(
-    title: str = "RoboSage App",
-    description: str = "Your robot, your voice, monetized."
-) -> gr.Blocks:
-    with gr.Blocks() as demo:
-        gr.Markdown(f"# 🚀 {title}\n\n{description}")
-
-        # 1. Generate & Download Artifacts
-        with gr.Accordion("🎨 Generate App & Download Artifacts", open=True):
-            idea = gr.Textbox(label="Robot Idea", placeholder="e.g. A friendly greeting robot.")
-            gen_btn = gr.Button("Generate & Package", key="gen-app-btn")
-            status = gr.Textbox(label="Generation Status", interactive=False)
-            download_zip = gr.File(label="Download Artifacts (.zip)")
-            blueprint_view = gr.JSON(label="App Blueprint")
-            code_view = gr.Code(label="Generated Code Preview", language="python")
+    title = assets.get("blueprint", {}).get("title", "<unknown>")
+    return title
+
+# Synchronous wrapper for Gradio
+def generate_app(idea: str) -> str:
+    title = asyncio.run(run_pipeline_async(idea))
+    return f"✅ Generated app: {title}"
+
+# Package artifacts into a ZIP and return the path
+def get_app_zip() -> str:
+    # package_artifacts should return the absolute path to the ZIP
+    zip_path = package_artifacts()
+    if not os.path.isfile(zip_path):
+        raise FileNotFoundError(f"Expected ZIP at {zip_path}, but not found.")
+    return zip_path
+
+# Simulated robot behavior
+def robot_behavior(command: str) -> str:
+    return VirtualRobot().perform_action(command)
+
+# Launch the Gradio interface
+def launch_gradio_app():
+    with gr.Blocks(css=".gradio-container { width: 800px; margin: auto; }
+                   .section { margin-bottom: 20px; }") as demo:
+        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
+
+        with gr.Group(elem_id="generate-app"), gr.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_output = gr.Textbox(label="Status", interactive=False)
+
             gen_btn.click(
-                fn=on_generate,
-                inputs=[idea],
-                outputs=[status, download_zip, blueprint_view, code_view]
+                fn=generate_app,
+                inputs=[idea_input],
+                outputs=[status_output]
+            )
+
+        with gr.Group(elem_id="download-app"), gr.Box():
+            gr.Markdown("## 📦 Download Packaged App")
+            dl_btn = gr.Button("Download App ZIP")
+            dl_file = gr.File(label="App ZIP File")
+
+            dl_btn.click(
+                fn=get_app_zip,
+                inputs=None,
+                outputs=[dl_file]
+            )
+
+        with gr.Group(elem_id="robot-sim"), gr.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)
 
-            # Browse artifacts directory
-            gr.Markdown("**Browse Generated Artifacts**")
-            files_browser = gr.File(
-                label="Artifacts Directory",
-                file_count="multiple",
-                file_types=None,
-                value="/data"
+            sim_btn.click(
+                fn=robot_behavior,
+                inputs=[cmd_input],
+                outputs=[sim_output]
             )
 
-        # 2. Robot Simulation (Text & Voice)
-        with gr.Accordion("🤖 Test Your Robot", open=False):
-            text_input = gr.Textbox(label="Text Command", placeholder="Type 'hello' or 'say Good job!'" )
-            text_btn = gr.Button("Send Text", key="send-text-btn")
-            text_output = gr.Textbox(label="Robot Response", lines=3, interactive=False)
-            text_btn.click(fn=robot_behavior, inputs=[text_input], outputs=[text_output])
-
-            gr.Markdown("---")
-
-            audio_input = gr.Audio(source="microphone", type="filepath", label="Record Command")
-            audio_btn = gr.Button("Send Audio", key="send-audio-btn")
-            audio_output = gr.Textbox(label="Robot Response (via voice)", lines=3, interactive=False)
-            audio_btn.click(fn=transcribe_and_respond, inputs=[audio_input], outputs=[audio_output])
-
-        # 3. Monetization & Revenue Dashboard
-        with gr.Accordion("💰 Monetization Dashboard", open=False):
-            subscribe_btn = gr.Button("Subscribe to Pro Plan", key="subscribe-btn")
-            subscribe_msg = gr.Textbox(label="Subscription Status", interactive=False)
-            rev_btn = gr.Button("View Revenue Stats", key="rev-stats-btn")
-            rev_table = gr.Dataframe(label="Revenue & Usage Metrics")
-            subscribe_btn.click(fn=lambda: "✅ Subscribed!", inputs=None, outputs=[subscribe_msg])
-            rev_btn.click(fn=get_revenue_stats, inputs=None, outputs=[rev_table])
+        demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), share=False)
 
     return demo
 
-if __name__ == "__main__":
-    app = launch_gradio_app()
-    app.launch()
+
+# Expose interface for external imports
+robot_behavior = robot_behavior
+launch_gradio_app = launch_gradio_app