Update app.py

app.py

@@ -1,7 +1,6 @@
 from fastapi.responses import JSONResponse, StreamingResponse, Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, HTTPException, Request
-import mimetypes
 from threading import Thread
 from Instance import Instance
 from api import LoadBalancerAPI
@@ -32,18 +31,25 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Mapping of file extensions to MIME types
+MIME_TYPES = {
+    ".mp4": "video/mp4",
+    ".webm": "video/webm",
+    ".mkv": "video/webm",
+    ".ts": "video/mp2t",
+}
+
 async def serve_video(file_path: str, request: Request):
     """Serve video file with support for range requests and proper CORS headers."""
     if not os.path.isfile(file_path):
         raise HTTPException(status_code=404, detail="Video file not found")
-    
+
     file_size = os.path.getsize(file_path)
     range_header = request.headers.get('range', None)
 
     # Determine the MIME type based on the file extension
-    mime_type, _ = mimetypes.guess_type(file_path)
-    if mime_type is None:
-        mime_type = 'application/octet-stream'  # Fallback MIME type
+    file_extension = os.path.splitext(file_path)[1].lower()
+    mime_type = MIME_TYPES.get(file_extension, 'application/octet-stream')  # Default to octet-stream if no match
 
     # CORS headers to attach to all responses
     cors_headers = {
@@ -132,28 +138,28 @@ async def get_movie_api(request: Request, title: str):
     """Endpoint to get the movie by title with support for range requests."""
     if not title:
         raise HTTPException(status_code=400, detail="Title parameter is required")
-    
+
     # Check if the film is already cached
     if title in instance.FILM_STORE:
         cache_path = instance.FILM_STORE[title]
         if os.path.exists(cache_path):
             return await serve_video(cache_path, request)
-    
+
     movie_path = instance.find_movie_path(title)
-    
+
     if not movie_path:
         raise HTTPException(status_code=404, detail="Movie not found")
-    
+
     cache_path = os.path.join(CACHE_DIR, movie_path)
     file_url = f"https://huggingface.co/{REPO}/resolve/main/{movie_path}"
     film_id = instance.get_film_id(title)
-    
+
     # Start the download in a separate thread if not already downloading
     if film_id not in instance.download_threads or not instance.download_threads[film_id].is_alive():
         thread = Thread(target=instance.download_film, args=(file_url, TOKEN, cache_path, film_id, title))
         instance.download_threads[film_id] = thread
         thread.start()
-    
+
     return JSONResponse({"status": "Download started", "film_id": film_id})
 
 @app.get("/api/get/tv/{title}/{season}/{episode}")
@@ -171,7 +177,7 @@ async def get_tv_show_api(request: Request, title: str, season: str, episode: st
                     return await serve_video(cache_path, request)
 
     tv_path = instance.find_tv_path(title)
-    
+
     if not tv_path:
         raise HTTPException(status_code=404, detail="TV show not found")
 
@@ -189,21 +195,21 @@ async def get_tv_show_api(request: Request, title: str, season: str, episode: st
 
     if not episode_path:
         raise HTTPException(status_code=404, detail="Episode not found")
-    
+
     cache_path = os.path.join(CACHE_DIR, episode_path)
     file_url = f"https://huggingface.co/{REPO}/resolve/main/{episode_path}"
     episode_id = instance.encode_episodeid(title, season, episode)
-    
+
     # Start the download in a separate thread if not already downloading
     if episode_id not in instance.download_threads or not instance.download_threads[episode_id].is_alive():
         thread = Thread(target=instance.download_episode, args=(file_url, TOKEN, cache_path, episode_id, title))
         instance.download_threads[episode_id] = thread
         thread.start()
-    
+
     return JSONResponse({"status": "Download started", "episode_id": episode_id})
 
 @app.get("/api/get/progress/{id}")
 async def get_progress_api(id: str):
     """Endpoint to get the download progress of a movie or TV show episode."""
     progress = instance.get_download_progress(id)
-    return JSONResponse({"id": id, "progress": progress})
+    return JSONResponse({"id": id, "progress": progress})