Update app.py

app.py

@@ -1,11 +1,67 @@
+# from fastapi import FastAPI
+# from fastapi.responses import FileResponse
+# import os
+# import subprocess
+# from manim import *
+
+# app = FastAPI()
+
+# @app.get("/")
+# def read_root():
+#     return {"message": "Welcome to the Manim API on HuggingFace!"}
+
+# @app.get("/generate")
+# def generate_animation():
+#     class PointMovingOnShapes(Scene):
+#         def construct(self):
+#             # Define the circle and dot
+#             circle = Circle(radius=1, color=BLUE)
+#             dot = Dot()
+#             dot2 = dot.copy().shift(RIGHT)
+#             self.add(dot)
+
+#             # Create a line
+#             line = Line([3, 0, 0], [5, 0, 0])
+#             self.add(line)
+
+#             # Animate the circle, dot, and line
+#             self.play(GrowFromCenter(circle))
+#             self.play(Transform(dot, dot2))
+#             self.play(MoveAlongPath(dot, circle), run_time=2, rate_func=linear)
+#             self.play(Rotating(dot, about_point=[2, 0, 0]), run_time=1.5)
+#             self.wait()
+
+#     # Directory to save rendered video
+#     output_dir = "/tmp/manim_output"
+#     os.makedirs(output_dir, exist_ok=True)
+
+#     # Rendering the scene to a video file
+#     scene = PointMovingOnShapes()
+#     scene.render(directory=output_dir)
+
+#     # Path to the rendered video file (manim by default saves as .mp4)
+#     video_path = os.path.join(output_dir, "media", "videos", "PointMovingOnShapes", "1080p60", "PointMovingOnShapes.mp4")
+
+#     # Return the video file as a response
+#     if os.path.exists(video_path):
+#         return FileResponse(video_path)
+#     else:
+#         return {"message": "Error: Animation rendering failed."}
+
+
 from fastapi import FastAPI
-from fastapi.responses import FileResponse
+from fastapi.responses import JSONResponse
 import os
-import subprocess
 from manim import *
+from pyuploadcare import Uploadcare
+import shutil
 
 app = FastAPI()
 
+UPLOADCARE_PUBLIC_KEY = os.environ.get("key")
+UPLOADCARE_SECRET_KEY = os.environ.get("secret")
+uploadcare = Uploadcare(public_key=UPLOADCARE_PUBLIC_KEY, secret_key=UPLOADCARE_SECRET_KEY)
+
 @app.get("/")
 def read_root():
     return {"message": "Welcome to the Manim API on HuggingFace!"}
@@ -39,11 +95,21 @@ def generate_animation():
     scene = PointMovingOnShapes()
     scene.render(directory=output_dir)
 
-    # Path to the rendered video file (manim by default saves as .mp4)
+    # Path to the rendered video file (Manim saves as .mp4)
     video_path = os.path.join(output_dir, "media", "videos", "PointMovingOnShapes", "1080p60", "PointMovingOnShapes.mp4")
 
-    # Return the video file as a response
     if os.path.exists(video_path):
-        return FileResponse(video_path)
+        try:
+            # Upload the video file to Uploadcare
+            uploaded_file = uploadcare.upload(video_path)
+            
+            # Get the CDN URL of the uploaded file
+            file_url = uploaded_file.cdn_url
+
+            # Return the URL as a JSON response
+            return JSONResponse(content={"file_url": file_url})
+        except Exception as e:
+            return JSONResponse(content={"error": f"Upload failed: {str(e)}"}, status_code=500)
     else:
-        return {"message": "Error: Animation rendering failed."}
+        return JSONResponse(content={"message": "Error: Animation rendering failed."}, status_code=500)
+