Update app.py

app.py

@@ -254,39 +254,52 @@ Remember: Simpler is better. Only use advanced ffmpeg features if absolutely nec
             max_tokens=2048,
         )
         content = completion.choices[0].message.content
+        print(f"\n=== RAW API RESPONSE ===\n{content}\n========================\n")
+        
         # Extract command from code block if present
-        if "```" in content:
-            # Find content between ```sh or ```bash and the next ```
-            import re
-
-            command_match = re.search(
-                r"```(?:sh|bash)?\n(.*?)\n```", content, re.DOTALL
-            )
-            if command_match:
-                command = command_match.group(1).strip()
-            else:
-                # Try to find a line that starts with ffmpeg
-                ffmpeg_lines = [
-                    line.strip()
-                    for line in content.split("\n")
-                    if line.strip().startswith("ffmpeg")
-                ]
-                if ffmpeg_lines:
-                    command = ffmpeg_lines[0]
-                else:
-                    command = content.replace("\n", "")
-        else:
-            # Try to find a line that starts with ffmpeg
+        import re
+        command = None
+        
+        # Try multiple code block patterns
+        code_patterns = [
+            r"```(?:bash|sh|shell)?\n(.*?)\n```",  # Standard code blocks
+            r"```\n(.*?)\n```",  # Plain code blocks
+            r"`([^`]*ffmpeg[^`]*)`",  # Inline code with ffmpeg
+        ]
+        
+        for pattern in code_patterns:
+            matches = re.findall(pattern, content, re.DOTALL | re.IGNORECASE)
+            for match in matches:
+                if "ffmpeg" in match.lower():
+                    command = match.strip()
+                    break
+            if command:
+                break
+        
+        # If no code block found, try to find ffmpeg lines directly
+        if not command:
             ffmpeg_lines = [
                 line.strip()
                 for line in content.split("\n")
-                if line.strip().startswith("ffmpeg")
+                if line.strip().lower().startswith("ffmpeg")
             ]
             if ffmpeg_lines:
                 command = ffmpeg_lines[0]
-            else:
-                command = content.replace("\n", "")
-
+        
+        # Last resort: look for any line containing ffmpeg
+        if not command:
+            for line in content.split("\n"):
+                line = line.strip()
+                if "ffmpeg" in line.lower() and len(line) > 10:
+                    command = line
+                    break
+        
+        if not command:
+            print(f"ERROR: No ffmpeg command found in response")
+            command = content.replace("\n", " ").strip()
+        
+        print(f"=== EXTRACTED COMMAND ===\n{command}\n========================\n")
+        
         # remove output.mp4 with the actual output file path
         command = command.replace("output.mp4", "")