robosage / app.py
mgbam's picture
Update app.py
0c77c4b verified
raw
history blame
1.83 kB
import os
import asyncio
import gradio as gr
from deployer.gradio_generator import deploy_callback, robot_behavior
from deployer.gradio_generator import launch_gradio_app
# Wrapper to run deploy_callback asynchronously
async def _deploy_async(idea: str):
return deploy_callback(idea)
# Synchronous Gradio callback for app generation
def generate_app(idea: str):
status, zip_path = asyncio.run(_deploy_async(idea))
return status, zip_path
# Build and launch the Gradio interface
def main():
with gr.Blocks(css=".gradio-container { max-width: 900px; margin: auto; }
.section { padding: 1rem; }") as demo:
gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
with gr.Row():
with gr.Column(scale=1):
gr.Markdown("## 1️⃣ Generate & Download App")
idea_input = gr.Textbox(label="Your Robot Idea", placeholder="e.g. A friendly greeting robot.", lines=2)
gen_btn = gr.Button("Generate App")
status_out = gr.Textbox(label="Status", interactive=False)
zip_out = gr.File(label="Download App ZIP")
gen_btn.click(fn=generate_app, inputs=[idea_input], outputs=[status_out, zip_out])
with gr.Column(scale=1):
gr.Markdown("## 2️⃣ Robot Simulator")
cmd_input = gr.Textbox(label="Command", placeholder="hello or say You rock!", lines=1)
sim_btn = gr.Button("Send Command")
sim_out = gr.Textbox(label="Robot Response", interactive=False)
sim_btn.click(fn=robot_behavior, inputs=[cmd_input], outputs=[sim_out])
demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), share=False)
if __name__ == "__main__":
main()