Temporary patch before tomorrow

app.py

@@ -22,59 +22,51 @@ def generate(*args):
             im.save(p.stdin, "PNG")
         p.stdin.close()
         p.wait()
-    return video_path
+    model_path = None  # TODO
+    return [video_path, model_path]
 
 
-def main():
-    with gr.Blocks() as ui:
-        gr.Markdown("# Pulsar+CLIP")
-        gr.Markdown("Generate 3D point clouds from text!")
-        
-        with gr.Group():
-            gr.Markdown("## Settings")
-            inputs = []
-            defaults = []
-            with gr.Tabs():
-                for name, section in CONFIG_SPEC:
-                    with gr.TabItem(name):
-                        for k, v0, t in section:
-                            if t in (float, int):
-                                element = gr.Number(label=k, value=v0)
-                            elif t == str:
-                                element = gr.Textbox(label=k, value=v0)
-                            elif t == bool:
-                                element = gr.Checkbox(label=k, value=v0)
-                            elif isinstance(t, tuple):
-                                element = gr.Slider(*t, label=k, value=v0)
-                            elif isinstance(t, list):
-                                element = gr.Dropdown(label=k, value=v0, choices=t)
-                            else:
-                                raise TypeError(f"Input format {t} should be one of str, int, bool, tuple, list")
-                                element = 1/0
-                            inputs.append(element)
-                            defaults.append(v0)
-        
-        with gr.Row():
-            with gr.Column():
-                button = gr.Button("Run")
-                gr.Markdown("## Result")
-                output = gr.Video()
-                
-                button.click(fn=generate, inputs=inputs, outputs=output)
-        
-            with gr.Column():
-                gr.Markdown("## Examples")
-                gr.Examples(fn=generate, inputs=inputs, outputs=output,
-                            examples=[defaults])
+with gr.Blocks() as ui:
+    gr.Markdown("# Pulsar+CLIP")
+    gr.Markdown("Generate 3D point clouds from text!")
     
-    ui.launch()
+    with gr.Group():
+        gr.Markdown("## Settings")
+        inputs = []
+        defaults = []
+        with gr.Tabs():
+            for name, section in CONFIG_SPEC:
+                with gr.TabItem(name):
+                    for k, v0, t in section:
+                        if t in (float, int):
+                            element = gr.Number(label=k, value=v0)
+                        elif t == str:
+                            element = gr.Textbox(label=k, value=v0)
+                        elif t == bool:
+                            element = gr.Checkbox(label=k, value=v0)
+                        elif isinstance(t, tuple):
+                            element = gr.Slider(*t, label=k, value=v0)
+                        elif isinstance(t, list):
+                            element = gr.Dropdown(label=k, value=v0, choices=t)
+                        else:
+                            raise TypeError(f"Input format {t} should be one of str, int, bool, tuple, list")
+                            element = 1/0
+                        inputs.append(element)
+                        defaults.append(v0)
     
-    #gr.Interface(inputs=[
-        #(gr.inputs.Number(label=k, default=v0) if t in (float, int) else
-         #gr.inputs.Checkbox(label=k, default=v0) if t == bool else gr.inputs.Textbox(label=k, default=v0) if t == str
-         #else gr.inputs.Dropdown(label=k, default=v0, choices=t) if isinstance(t, (tuple, list)) else 1/0)
-        #for k, v0, t in CONFIG_SPEC], outputs=gr.outputs.Video(), fn=generate).launch()
-
+    button = gr.Button("Run")
+    gr.Markdown("## Result")
+    with gr.Row():
+        with gr.Column():
+            video = gr.Video()
+        with gr.Column():
+            model = gr.Model3D()
+    
+    button.click(fn=generate, inputs=inputs, outputs=[video, model])
+    
+    gr.Markdown("## Examples")
+    gr.Examples(fn=generate, inputs=inputs, outputs=[video, model],
+                examples=[defaults], cache_examples=True, examples_per_page=1)
 
-if __name__ == '__main__':
-    main()
+ui.launch()
+demo = ui