Update app.py

app.py

@@ -81,13 +81,16 @@ def infer(text):
         return "Error in motion generation."
 
     try:
-        gif_data_list = draw_to_batch([motion_data], [text], None)
-        gif_bytes = gif_data_list[0]
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".gif") as temp_file:
-            temp_file.write(gif_bytes)
-            temp_file_path = temp_file.name
-        print("GIF successfully saved to temporary file.")
-        return temp_file_path
+        gif_data = draw_to_batch([motion_data], [text], None)
+        if gif_data:
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".gif") as temp_file:
+                temp_file.write(gif_data)
+                temp_file_path = temp_file.name
+            print("GIF successfully saved to temporary file.")
+            return temp_file_path
+        else:
+            print("Failed to generate GIF data.")
+            return "Error generating GIF. Please try again."
     except Exception as e:
         print(f"Error generating GIF: {str(e)}")
         return "Error generating GIF. Please try again."
@@ -105,5 +108,6 @@ with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
             outputs=[output_image]
         )
 
+
 if __name__ == "__main__":
     demo.launch()