input validation

app.py

@@ -69,6 +69,12 @@ class DemoRunner:
         return lmp_tabletop_ui
 
     def setup(self, api_key, model_name, n_blocks, n_bowls):
+        if not api_key:
+            return 'Please enter your OpenAI API key!', None
+        
+        if n_blocks + n_bowls == 0:
+            return 'Please select at least one object!', None
+
         openai.api_key = api_key
         self._model_name = model_name
 
@@ -87,16 +93,15 @@ class DemoRunner:
 
     def run(self, instruction):
         if self._env is None:
-            return 'Please run setup first'
+            return 'Please run setup first!', None, None
 
         self._env.cache_video = []
         self._md_logger.clear()
 
         try:
             self._lmp_tabletop_ui(instruction, f'objects = {self._env.object_list}')
-            run_info = self._md_logger.get_log()
         except Exception as e:
-            run_info = f'Error: {e}'
+            return str(e), None, None
 
         video_file_name = None
         if self._env.cache_video:
@@ -104,7 +109,7 @@ class DemoRunner:
             video_file_name = NamedTemporaryFile(suffix='.mp4').name
             rendered_clip.write_videofile(video_file_name, fps=25)
 
-        return run_info, self._env.get_camera_image(), video_file_name
+        return self._md_logger.get_log(), self._env.get_camera_image(), video_file_name
 
 
 if __name__ == '__main__':
@@ -129,7 +134,7 @@ if __name__ == '__main__':
                     inp_n_bowls = gr.Slider(label='Number of Bowls', minimum=0, maximum=4, value=3, step=1)
             
                 btn_setup = gr.Button("Setup/Reset Env")
-                info_setup = gr.Markdown(label='Setup Info', value='## Available Objects:')
+                info_setup = gr.Markdown(label='Setup Info')
             with gr.Column():
                 img_setup = gr.Image(label='Current Simulation')