adjusted a few things

Author: yeshaokai

amadeusgpt/managers/visual_manager.py

@@ -549,7 +549,7 @@ def write_video(self, out_folder, video_file_path, out_name, events):
         if total_duration < 0.0:
             return
 
-        fourcc = cv2.VideoWriter_fourcc(*"avc1")  # Adjust the codec as needed
+        fourcc = cv2.VideoWriter_fourcc(*"h264")  # Adjust the codec as needed
 
         out_videos = []        
 
@@ -667,8 +667,7 @@ def write_video(self, out_folder, video_file_path, out_name, events):
                         )                 
                     out.write(frame)
                 offset += 1
-                if current_frame == time_slice[1]:
-                    
+                if current_frame == time_slice[1]:                    
                     out.release()
                     break
 

notebook/EPM_demo.ipynb

@@ -115,14 +115,21 @@
    "outputs": [],
    "source": [
     "from IPython.display import Markdown, display\n",
+    "from IPython.display import Video\n",
     "def parse_result(amadeus, qa_message):\n",
     "    display(Markdown(qa_message['chain_of_thought']))\n",
     "    sandbox = amadeus.sandbox\n",
     "    qa_message = sandbox.code_execution(qa_message)\n",
     "    sandbox.render_qa_message(qa_message)\n",
     "    print ('after executing the function')\n",
     "    display(qa_message['meta_info'])\n",
-    "    print(qa_message['pose_video'])\n",
+    "    print (f'videos generated to {qa_message[\"out_videos\"]}')\n",
+    "    print ('Open it with media player if it does not properly display in the notebook')\n",
+    "    if qa_message['out_videos'] is not None:\n",
+    "        for video_path in qa_message['out_videos']:\n",
+    "\n",
+    "            display(Video(video_path, embed=True))\n",
+    "\n",
     "    display(Markdown(str(qa_message['function_rets'])))"
    ]
   },

notebook/Horse_demo.ipynb

@@ -115,14 +115,21 @@
    "outputs": [],
    "source": [
     "from IPython.display import Markdown, display\n",
+    "from IPython.display import Video\n",
     "def parse_result(amadeus, qa_message):\n",
     "    display(Markdown(qa_message['chain_of_thought']))\n",
     "    sandbox = amadeus.sandbox\n",
     "    qa_message = sandbox.code_execution(qa_message)\n",
     "    sandbox.render_qa_message(qa_message)\n",
     "    print ('after executing the function')\n",
     "    display(qa_message['meta_info'])\n",
-    "    print(qa_message['pose_video'])\n",
+    "    print (f'videos generated to {qa_message[\"out_videos\"]}')\n",
+    "    print ('Open it with media player if it does not properly display in the notebook')\n",
+    "    if qa_message['out_videos'] is not None:\n",
+    "        for video_path in qa_message['out_videos']:\n",
+    "\n",
+    "            display(Video(video_path, embed=True))\n",
+    "\n",
     "    display(Markdown(str(qa_message['function_rets'])))"
    ]
   },

notebook/MABe_demo.ipynb

@@ -119,14 +119,21 @@
    "outputs": [],
    "source": [
     "from IPython.display import Markdown, display\n",
+    "from IPython.display import Video\n",
     "def parse_result(amadeus, qa_message):\n",
     "    display(Markdown(qa_message['chain_of_thought']))\n",
     "    sandbox = amadeus.sandbox\n",
     "    qa_message = sandbox.code_execution(qa_message)\n",
     "    sandbox.render_qa_message(qa_message)\n",
     "    print ('after executing the function')\n",
     "    display(qa_message['meta_info'])\n",
-    "    print(qa_message['pose_video'])\n",
+    "    print (f'videos generated to {qa_message[\"out_videos\"]}')\n",
+    "    print ('Open it with media player if it does not properly display in the notebook')\n",
+    "    if qa_message['out_videos'] is not None:\n",
+    "        for video_path in qa_message['out_videos']:\n",
+    "\n",
+    "            display(Video(video_path, embed=True))\n",
+    "\n",
     "    display(Markdown(str(qa_message['function_rets'])))"
    ]
   },