|
|
import os |
|
|
import gradio as gr |
|
|
|
|
|
from deployer.gradio_generator import deploy_callback, robot_behavior |
|
|
|
|
|
def main(): |
|
|
|
|
|
with gr.Blocks(css=""" |
|
|
.gradio-container { max-width: 900px; margin: auto; } |
|
|
.generate-section, .simulate-section { padding: 1rem; } |
|
|
""") as demo: |
|
|
|
|
|
gr.Markdown("# 🚀 RoboSage\nGenerate & download a simple robot app, then test it live.") |
|
|
|
|
|
with gr.Column(elem_id="generate-section"): |
|
|
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 & ZIP") |
|
|
status_out = gr.Textbox(label="Status", interactive=False) |
|
|
zip_out = gr.File(label="Download App ZIP") |
|
|
|
|
|
gen_btn.click( |
|
|
fn=deploy_callback, |
|
|
inputs=[idea_input], |
|
|
outputs=[status_out, zip_out] |
|
|
) |
|
|
|
|
|
with gr.Column(elem_id="simulate-section"): |
|
|
gr.Markdown("## 2️⃣ Robot Simulator") |
|
|
cmd_input = gr.Textbox( |
|
|
label="Command", |
|
|
placeholder="say 'hello' or '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() |
|
|
|
