Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,24 +1,22 @@
-# gradio_generator.py - CPU & Spaces-compatible Gradio interface with full monetization & artifact management for RoboSage
+# gradio_generator.py - CPU & Spaces-compatible Gradio interface with HF Inference API for STT & Monetization
 
 import os
 import json
+import requests
 import gradio as gr
-from transformers import pipeline
 from deployer.simulator_interface import VirtualRobot
 from deployer.revenue_tracker import get_revenue_stats, package_artifacts
 from core_creator.voice_to_app import VoiceToAppCreator
 
-# Initialize Hugging Face ASR pipeline
+# 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 audio transcription via HuggingFace API.")
-asr = pipeline(
-    task="automatic-speech-recognition",
-    model="openai/whisper-small",
-    use_auth_token=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 actions
+# Core robot logic
 def robot_behavior(user_input: str) -> str:
     bot = VirtualRobot()
     text = user_input.strip().lower()
@@ -28,27 +26,36 @@ def robot_behavior(user_input: str) -> str:
         return bot.perform_action(text)
     return bot.perform_action("say I'm sorry, I didn't understand that.")
 
-# Transcription helper
+# Transcribe via Hugging Face Inference API
 def transcribe_audio(audio_file: str) -> str:
-    result = asr(audio_file)
+    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()
+    # The inference API returns {'text': '...'}
     return result.get("text", "")
 
+# Combined flow
 def transcribe_and_respond(audio_file: str) -> str:
     text = transcribe_audio(audio_file)
     return robot_behavior(text)
 
-# Generation & packaging pipeline
-def generate_and_package_app(idea: str):
-    creator = VoiceToAppCreator(idea)
+# Package generation pipeline
+def on_generate(i: str):
+    creator = VoiceToAppCreator(i)
     assets = creator.run_pipeline()
-    # Package code & config into zip and return path
     zip_path = package_artifacts(assets)
-    # Prepare blueprint and code previews
     blueprint = assets.get("blueprint", {})
-    code_files = assets.get("code", {})  # dict of {filename: code_str}
-    # Concatenate code files for preview
+    code_files = assets.get("code", {})
     code_preview = "\n\n".join([f"# {fname}\n{content}" for fname, content in code_files.items()])
-    return zip_path, blueprint, code_preview
+    return (
+        "✅ App generated successfully!",
+        zip_path,
+        blueprint,
+        code_preview
+    )
 
 # Build full-featured Gradio app
 def launch_gradio_app(
@@ -58,7 +65,7 @@ def launch_gradio_app(
     with gr.Blocks() as demo:
         gr.Markdown(f"# 🚀 {title}\n\n{description}")
 
-        # 1️⃣ Generate & Download Artifacts
+        # 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")
@@ -66,54 +73,34 @@ def launch_gradio_app(
             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")
-
-            def on_generate(i):
-                zip_path, blueprint, code_preview = generate_and_package_app(i)
-                return (
-                    "✅ App generated successfully!", 
-                    zip_path, 
-                    blueprint, 
-                    code_preview
-                )
-
             gen_btn.click(
                 fn=on_generate,
                 inputs=[idea],
                 outputs=[status, download_zip, blueprint_view, code_view]
             )
 
-        # 2️⃣ Robot Simulation
-        with gr.Accordion("🤖 Test Your Robot (Text & Voice)", open=False):
-            # Text command
-            text_input = gr.Textbox(label="Text Command", placeholder="Type 'hello' or 'say Great job!'" )
+        # 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("---")
-            # Voice command
+
             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
+        # 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")
-
-            def on_subscribe():
-                # Stubbed subscription activation
-                return "✅ Subscribed! You now have access to premium features."
-
-            def on_view_revenue():
-                df = get_revenue_stats()
-                return df
-
-            subscribe_btn.click(fn=on_subscribe, inputs=None, outputs=[subscribe_msg])
-            rev_btn.click(fn=on_view_revenue, inputs=None, outputs=[rev_table])
+            subscribe_btn.click(fn=lambda: "✅ Subscribed!", inputs=None, outputs=[subscribe_msg])
+            rev_btn.click(fn=get_revenue_stats, inputs=None, outputs=[rev_table])
 
     return demo