File size: 2,018 Bytes
67f9d7b
660b656
43d80bd
f69fdac
1b23b17
 
 
 
8313465
1b23b17
f69fdac
1b23b17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38c7720
 
1b23b17
 
38c7720
f4f0212
0180e60
67f9d7b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import os
import gradio as gr
from deployer.gradio_generator import deploy_callback, robot_behavior

def generate_app(idea: str):
    """
    Gradio callback: returns (status_message, path_to_zip).
    Gradio's File component will serve the zip.
    """
    return deploy_callback(idea)

def main():
    with gr.Blocks() as demo:
        gr.Markdown("# 🚀 RoboSage\nGenerate & download a simple robot app, then test it live.")

        with gr.Row():
            # ▶️ Generate & Download App
            with gr.Column():
                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=generate_app,
                    inputs=[idea_input],
                    outputs=[status_out, zip_out],
                )

            # ▶️ Robot Simulator
            with gr.Column():
                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],
                )

        # Launch on Hugging Face Spaces or locally
        demo.launch(
            server_name="0.0.0.0",
            server_port=int(os.environ.get("PORT", 7860)),
            share=False,
        )

if __name__ == "__main__":
    main()