fix

app.py

@@ -1,5 +1,6 @@
-from fastapi import FastAPI
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, FileResponse
+from fastapi import FastAPI, HTTPException
+from threading import Thread
 from Instance import Instance
 from api import LoadBalancerAPI
 import os
@@ -34,3 +35,84 @@ async def get_tv_store_api():
 async def get_film_store_api():
     """Endpoint to get the TV store JSON."""
     return JSONResponse(instance.FILM_STORE)
+
+@app.get("/api/get/film/{title}")
+async def get_movie_api(title: str):
+    """Endpoint to get the movie by title."""
+    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 FileResponse(cache_path)
+    
+    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}")
+async def get_tv_show_api(title: str, season: str, episode: str):
+    """Endpoint to get the TV show by title, season, and episode."""
+    if not title or not season or not episode:
+        raise HTTPException(status_code=400, detail="Title, season, and episode parameters are required")
+
+    # Check if the episode is already cached
+    if title in instance.TV_STORE and season in instance.TV_STORE[title]:
+        for ep in instance.TV_STORE[title][season]:
+            if episode in ep:
+                cache_path = instance.TV_STORE[title][season][ep]
+                if os.path.exists(cache_path):
+                    return FileResponse(cache_path)
+
+    tv_path = instance.find_tv_path(title)
+    
+    if not tv_path:
+        raise HTTPException(status_code=404, detail="TV show not found")
+
+    episode_path = None
+    for directory in instance.file_structure:
+        if directory['type'] == 'directory' and directory['path'] == 'tv':
+            for sub_directory in directory['contents']:
+                if sub_directory['type'] == 'directory' and title.lower() in sub_directory['path'].lower():
+                    for season_dir in sub_directory['contents']:
+                        if season_dir['type'] == 'directory' and season in season_dir['path']:
+                            for episode_file in season_dir['contents']:
+                                if episode_file['type'] == 'file' and episode in episode_file['path']:
+                                    episode_path = episode_file['path']
+                                    break
+
+    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})