Update deployer/gradio_generator.py
Browse files
deployer/gradio_generator.py
CHANGED
@@ -3,14 +3,16 @@ from .simulator_interface import VirtualRobot
|
|
3 |
import logging
|
4 |
|
5 |
class RoboSageApp:
|
6 |
-
def __init__(self):
|
7 |
self.robot = VirtualRobot()
|
|
|
|
|
8 |
self.logger = logging.getLogger(__name__)
|
9 |
self.interface = self._build_interface()
|
10 |
|
11 |
def _build_interface(self) -> gr.Blocks:
|
12 |
-
with gr.Blocks(title=
|
13 |
-
gr.Markdown("# 🤖
|
14 |
|
15 |
with gr.Row():
|
16 |
input_txt = gr.Textbox(label="Command", placeholder="Try 'wave' or 'say hello'")
|
@@ -27,12 +29,10 @@ class RoboSageApp:
|
|
27 |
|
28 |
return interface
|
29 |
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
return RoboSageApp().interface
|
34 |
|
35 |
-
# For local testing
|
36 |
if __name__ == "__main__":
|
37 |
logging.basicConfig(level=logging.INFO)
|
38 |
app = launch_gradio_app()
|
|
|
3 |
import logging
|
4 |
|
5 |
class RoboSageApp:
|
6 |
+
def __init__(self, title="RoboSage", description="Your virtual assistant"):
|
7 |
self.robot = VirtualRobot()
|
8 |
+
self.title = title
|
9 |
+
self.description = description
|
10 |
self.logger = logging.getLogger(__name__)
|
11 |
self.interface = self._build_interface()
|
12 |
|
13 |
def _build_interface(self) -> gr.Blocks:
|
14 |
+
with gr.Blocks(title=self.title) as interface:
|
15 |
+
gr.Markdown(f"# 🤖 {self.title}\n\n{self.description}")
|
16 |
|
17 |
with gr.Row():
|
18 |
input_txt = gr.Textbox(label="Command", placeholder="Try 'wave' or 'say hello'")
|
|
|
29 |
|
30 |
return interface
|
31 |
|
32 |
+
def launch_gradio_app(title="RoboSage", description="Your virtual assistant") -> gr.Blocks:
|
33 |
+
"""Now accepts title and description parameters"""
|
34 |
+
return RoboSageApp(title=title, description=description).interface
|
|
|
35 |
|
|
|
36 |
if __name__ == "__main__":
|
37 |
logging.basicConfig(level=logging.INFO)
|
38 |
app = launch_gradio_app()
|