Upload app.py

app.py

@@ -446,6 +446,19 @@ async def render_animation_endpoint(
         )
 
 
+@app.options("/media/{request_id}/{filename}", tags=["media"])
+async def download_file_options(request_id: str, filename: str):
+    """Handle CORS preflight requests for media files"""
+    return JSONResponse(
+        content={},
+        headers={
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "GET, HEAD, OPTIONS",
+            "Access-Control-Allow-Headers": "*",
+        }
+    )
+
+
 @app.get("/media/{request_id}/{filename}", tags=["media"])
 async def download_file(request_id: str, filename: str):
     """
@@ -494,7 +507,10 @@ async def download_file(request_id: str, filename: str):
         filename=filename,
         headers={
             "Cache-Control": "public, max-age=3600",
-            "Content-Disposition": f'inline; filename="{filename}"'
+            "Content-Disposition": f'inline; filename="{filename}"',
+            "Access-Control-Allow-Origin": "*",
+            "Access-Control-Allow-Methods": "GET, HEAD, OPTIONS",
+            "Access-Control-Allow-Headers": "*"
         }
     )
 
@@ -621,18 +637,18 @@ def get_render_log_tail(request_id: str, max_chars: int = 2000) -> Optional[str]
     if not log_path.exists():
         return None
 
-    try:
-        max_bytes = max_chars * 4
-        with log_path.open("rb") as handle:
-            handle.seek(0, os.SEEK_END)
-            size = handle.tell()
-            read_size = min(size, max_bytes)
-            handle.seek(-read_size, os.SEEK_END)
-            data = handle.read()
-        text = data.decode("utf-8", errors="replace").replace("\r", "\n")
-        if len(text) > max_chars:
-            text = text[-max_chars:]
-        return text.strip()
+    try:
+        max_bytes = max_chars * 4
+        with log_path.open("rb") as handle:
+            handle.seek(0, os.SEEK_END)
+            size = handle.tell()
+            read_size = min(size, max_bytes)
+            handle.seek(-read_size, os.SEEK_END)
+            data = handle.read()
+        text = data.decode("utf-8", errors="replace").replace("\r", "\n")
+        if len(text) > max_chars:
+            text = text[-max_chars:]
+        return text.strip()
     except OSError:
         return None
 
@@ -694,21 +710,21 @@ async def run_mcp_render(
         return False, error_message, {}
 
     # Convert to requested formats
-    formats_to_convert = []
-    output_format = arguments.get("output_format", "mp4")
-
-    if output_format == "all":
-        formats_to_convert = ["gif", "webm"]
-    elif output_format != "mp4":
-        formats_to_convert = [output_format]
-
-    conversion_results = {"mp4": output_mp4}
-    if formats_to_convert:
-        conversion_results.update(await convert_all_formats(
-            input_mp4=output_mp4,
-            output_formats=formats_to_convert,
-            request_id=request_id
-        ))
+    formats_to_convert = []
+    output_format = arguments.get("output_format", "mp4")
+
+    if output_format == "all":
+        formats_to_convert = ["gif", "webm"]
+    elif output_format != "mp4":
+        formats_to_convert = [output_format]
+
+    conversion_results = {"mp4": output_mp4}
+    if formats_to_convert:
+        conversion_results.update(await convert_all_formats(
+            input_mp4=output_mp4,
+            output_formats=formats_to_convert,
+            request_id=request_id
+        ))
 
     # Generate download URLs
     base_url = os.getenv("SPACE_HOST", "https://euler314-manim-mcp.hf.space")
@@ -894,78 +910,78 @@ Use `check_render_status` to see progress and get the output when ready."""
         # Calculate expiration
         expires_at = datetime.now(timezone.utc) + timedelta(hours=1)
 
-        # Format response for ChatGPT
-        url_list = "\n".join([f"  - {fmt.upper()}: {url}" for fmt, url in urls.items()])
-        log_tail = get_render_log_tail(request_id)
-        log_block = ""
-        if log_tail:
-            log_block = f"""
-Recent output:
-```
-{log_tail}
-```"""
-
-        response_text = f"""Animation rendered successfully.
-
-Scene: {scene_name}
-Quality: {arguments.get('quality', '720p')}
-Request ID: `{request_id}`
-
-Download URLs:
-{url_list}
-
-Files expire at: {expires_at.strftime('%Y-%m-%d %H:%M:%S UTC')}
-{log_block}
-
-Your animation is ready. Use the video player below or download via the links.
-"""
+        # Format response for ChatGPT
+        url_list = "\n".join([f"  - {fmt.upper()}: {url}" for fmt, url in urls.items()])
+        log_tail = get_render_log_tail(request_id)
+        log_block = ""
+        if log_tail:
+            log_block = f"""
+Recent output:
+```
+{log_tail}
+```"""
+
+        response_text = f"""Animation rendered successfully.
+
+Scene: {scene_name}
+Quality: {arguments.get('quality', '720p')}
+Request ID: `{request_id}`
+
+Download URLs:
+{url_list}
+
+Files expire at: {expires_at.strftime('%Y-%m-%d %H:%M:%S UTC')}
+{log_block}
+
+Your animation is ready. Use the video player below or download via the links.
+"""
 
         # Get base URL for resources
         resource_base_url = os.getenv("SPACE_HOST", "https://euler314-manim-mcp.hf.space")
         if not resource_base_url.startswith("http"):
             resource_base_url = f"https://{resource_base_url}"
 
-        content_items = [
-            {
-                "type": "text",
-                "text": response_text
-            }
-        ]
-
-        mp4_url = urls.get("mp4")
-        if mp4_url:
-            content_items.append({
-                "type": "resource",
-                "resource": {
-                    "uri": mp4_url,
-                    "mimeType": "video/mp4"
-                }
-            })
-
-        # Return with both text content and UI component
-        return {
-            "content": content_items + [{
-                "type": "resource",
-                "resource": {
-                    "uri": f"{resource_base_url}/mcp/resources/video-player",
-                    "mimeType": "text/html",
-                    "text": response_text
-                }
-            }],
-            "isError": False,
-            "_meta": {
-                "openai/outputTemplate": f"{resource_base_url}/mcp/resources/video-player"
-            },
-            "structuredContent": {
-                "urls": urls,
-                "scene_name": scene_name,
-                "quality": arguments.get("quality", "720p"),
-                "request_id": request_id,
-                "expires_at": expires_at.isoformat(),
-                "log_tail": log_tail,
-                "status": "ready"
-            }
-        }
+        content_items = [
+            {
+                "type": "text",
+                "text": response_text
+            }
+        ]
+
+        mp4_url = urls.get("mp4")
+        if mp4_url:
+            content_items.append({
+                "type": "resource",
+                "resource": {
+                    "uri": mp4_url,
+                    "mimeType": "video/mp4"
+                }
+            })
+
+        # Return with both text content and UI component
+        return {
+            "content": content_items + [{
+                "type": "resource",
+                "resource": {
+                    "uri": f"{resource_base_url}/mcp/resources/video-player",
+                    "mimeType": "text/html",
+                    "text": response_text
+                }
+            }],
+            "isError": False,
+            "_meta": {
+                "openai/outputTemplate": f"{resource_base_url}/mcp/resources/video-player"
+            },
+            "structuredContent": {
+                "urls": urls,
+                "scene_name": scene_name,
+                "quality": arguments.get("quality", "720p"),
+                "request_id": request_id,
+                "expires_at": expires_at.isoformat(),
+                "log_tail": log_tail,
+                "status": "ready"
+            }
+        }
 
     except Exception as e:
         logger.exception(f"MCP render error: {e}")
@@ -1004,22 +1020,22 @@ async def get_render_status_for_mcp(arguments: dict) -> dict:
             }
 
         # Check if expired
-        if await cleanup_service.check_if_expired(request_id):
-            return {
-                "content": [{
-                    "type": "text",
-                    "text": f"""Render expired.
+        if await cleanup_service.check_if_expired(request_id):
+            return {
+                "content": [{
+                    "type": "text",
+                    "text": f"""Render expired.
 
 Request ID: `{request_id}`
-
-This render has expired (files are deleted after 1 hour). You'll need to re-render the animation."""
-                }],
-                "isError": False,
-                "structuredContent": {
-                    "request_id": request_id,
-                    "status": "expired"
-                }
-            }
+
+This render has expired (files are deleted after 1 hour). You'll need to re-render the animation."""
+                }],
+                "isError": False,
+                "structuredContent": {
+                    "request_id": request_id,
+                    "status": "expired"
+                }
+            }
 
         status_info = read_render_status(request_id)
         if status_info:
@@ -1035,40 +1051,40 @@ Recent output:
 {log_tail}
 ```"""
 
-            if status == "error":
-                return {
-                    "content": [{
-                        "type": "text",
-                        "text": f"""Render failed.
-
-Request ID: `{request_id}`
-
-{message}{log_block}"""
-                    }],
-                    "isError": True,
-                    "structuredContent": {
-                        "request_id": request_id,
-                        "status": "error",
-                        "log_tail": log_tail
-                    }
-                }
-            if status == "processing":
-                return {
-                    "content": [{
-                        "type": "text",
-                        "text": f"""Render in progress.
-
-Request ID: `{request_id}`
-
-The render is still processing. Please wait and try again.{log_block}"""
-                    }],
-                    "isError": False,
-                    "structuredContent": {
-                        "request_id": request_id,
-                        "status": "processing",
-                        "log_tail": log_tail
-                    }
-                }
+            if status == "error":
+                return {
+                    "content": [{
+                        "type": "text",
+                        "text": f"""Render failed.
+
+Request ID: `{request_id}`
+
+{message}{log_block}"""
+                    }],
+                    "isError": True,
+                    "structuredContent": {
+                        "request_id": request_id,
+                        "status": "error",
+                        "log_tail": log_tail
+                    }
+                }
+            if status == "processing":
+                return {
+                    "content": [{
+                        "type": "text",
+                        "text": f"""Render in progress.
+
+Request ID: `{request_id}`
+
+The render is still processing. Please wait and try again.{log_block}"""
+                    }],
+                    "isError": False,
+                    "structuredContent": {
+                        "request_id": request_id,
+                        "status": "processing",
+                        "log_tail": log_tail
+                    }
+                }
 
         # Get render info
         render_info = await cleanup_service.get_render_info(request_id)
@@ -1086,38 +1102,38 @@ Recent output:
 ```
 {log_tail}
 ```"""
-                return {
-                    "content": [{
-                        "type": "text",
-                        "text": f"""Render in progress.
-
-Request ID: `{request_id}`
-
-The render is still processing. Please wait and try again.{log_block}"""
-                    }],
-                    "isError": False,
-                    "structuredContent": {
-                        "request_id": request_id,
-                        "status": "processing",
-                        "log_tail": log_tail
-                    }
-                }
-            else:
-                return {
-                    "content": [{
-                        "type": "text",
-                        "text": f"""Render not found.
+                return {
+                    "content": [{
+                        "type": "text",
+                        "text": f"""Render in progress.
 
 Request ID: `{request_id}`
-
-No render found with this ID. It may have expired or never existed."""
-                    }],
-                    "isError": True,
-                    "structuredContent": {
-                        "request_id": request_id,
-                        "status": "not_found"
-                    }
-                }
+
+The render is still processing. Please wait and try again.{log_block}"""
+                    }],
+                    "isError": False,
+                    "structuredContent": {
+                        "request_id": request_id,
+                        "status": "processing",
+                        "log_tail": log_tail
+                    }
+                }
+            else:
+                return {
+                    "content": [{
+                        "type": "text",
+                        "text": f"""Render not found.
+
+Request ID: `{request_id}`
+
+No render found with this ID. It may have expired or never existed."""
+                    }],
+                    "isError": True,
+                    "structuredContent": {
+                        "request_id": request_id,
+                        "status": "not_found"
+                    }
+                }
 
         # Render exists - provide details
         base_url = os.getenv("SPACE_HOST", "https://euler314-manim-mcp.hf.space")
@@ -1133,56 +1149,56 @@ No render found with this ID. It may have expired or never existed."""
         time_remaining = expires_at - datetime.now(timezone.utc)
         minutes_remaining = int(time_remaining.total_seconds() / 60)
 
-        url_list = "\n".join([f"  - {fmt.upper()}: {url}" for fmt, url in urls.items()])
-        log_tail = get_render_log_tail(request_id)
-        log_block = ""
-        if log_tail:
-            log_block = f"""
-Recent output:
-```
-{log_tail}
-```"""
-
-        status_text = f"""Render ready.
-
-Request ID: `{request_id}`
-Available Formats: {', '.join(render_info['formats'])}
-File Size: {render_info.get('size_bytes', 0) / 1024 / 1024:.2f} MB
-Expires In: {minutes_remaining} minutes
-
-Download URLs:
-{url_list}
-
-Expiration: {expires_at.strftime('%Y-%m-%d %H:%M:%S UTC')}
-{log_block}
-"""
-
-        content_items = [{
-            "type": "text",
-            "text": status_text
-        }]
-
-        mp4_url = urls.get("mp4")
-        if mp4_url:
-            content_items.append({
-                "type": "resource",
-                "resource": {
-                    "uri": mp4_url,
-                    "mimeType": "video/mp4"
-                }
-            })
-
-        return {
-            "content": content_items,
-            "isError": False,
-            "structuredContent": {
-                "request_id": request_id,
-                "status": "ready",
-                "urls": urls,
-                "expires_at": expires_at.isoformat(),
-                "log_tail": log_tail
-            }
-        }
+        url_list = "\n".join([f"  - {fmt.upper()}: {url}" for fmt, url in urls.items()])
+        log_tail = get_render_log_tail(request_id)
+        log_block = ""
+        if log_tail:
+            log_block = f"""
+Recent output:
+```
+{log_tail}
+```"""
+
+        status_text = f"""Render ready.
+
+Request ID: `{request_id}`
+Available Formats: {', '.join(render_info['formats'])}
+File Size: {render_info.get('size_bytes', 0) / 1024 / 1024:.2f} MB
+Expires In: {minutes_remaining} minutes
+
+Download URLs:
+{url_list}
+
+Expiration: {expires_at.strftime('%Y-%m-%d %H:%M:%S UTC')}
+{log_block}
+"""
+
+        content_items = [{
+            "type": "text",
+            "text": status_text
+        }]
+
+        mp4_url = urls.get("mp4")
+        if mp4_url:
+            content_items.append({
+                "type": "resource",
+                "resource": {
+                    "uri": mp4_url,
+                    "mimeType": "video/mp4"
+                }
+            })
+
+        return {
+            "content": content_items,
+            "isError": False,
+            "structuredContent": {
+                "request_id": request_id,
+                "status": "ready",
+                "urls": urls,
+                "expires_at": expires_at.isoformat(),
+                "log_tail": log_tail
+            }
+        }
 
     except Exception as e:
         logger.exception(f"Error checking render status: {e}")