update

app.py

@@ -13,12 +13,17 @@ app = Flask(__name__)
 CACHE_DIR = os.getenv("CACHE_DIR")
 INDEX_FILE = os.getenv("INDEX_FILE")
 TOKEN = os.getenv("TOKEN")
+FILM_STORE_JSON_PATH = os.path.join(CACHE_DIR, "film_store.json")
 download_threads = {}
 
 # Ensure CACHE_DIR exists
 if not os.path.exists(CACHE_DIR):
     os.makedirs(CACHE_DIR)
 
+if not os.path.exists(FILM_STORE_JSON_PATH):
+    with open(FILM_STORE_JSON_PATH, 'w') as json_file:
+        json.dump({}, json_file)
+
 # Index the file structure
 indexer()
 
@@ -94,6 +99,17 @@ def get_movie():
     if not title:
         return jsonify({"error": "Title parameter is required"}), 400
     
+    # Load the film store JSON
+    with open(FILM_STORE_JSON_PATH, 'r') as json_file:
+        film_store_data = json.load(json_file)
+
+    # Check if the film is already cached
+    if title in film_store_data:
+        cache_path = film_store_data[title]
+        if os.path.exists(cache_path):
+            return send_from_directory(os.path.dirname(cache_path), os.path.basename(cache_path))
+    
+    # If not cached, find the movie path in the index file
     json_data = load_json(INDEX_FILE)
     movie_path = find_movie_path(json_data, title)
     
@@ -101,20 +117,17 @@ def get_movie():
         return jsonify({"error": "Movie not found"}), 404
     
     cache_path = os.path.join(CACHE_DIR, movie_path)
-    if not os.path.exists(cache_path):
-        file_url = f"https://huggingface.co/Unicone-Studio/jellyfin_media/resolve/main/{movie_path}"
-        proxies = get_system_proxies()
-        film_id = get_film_id(title)
-        
-        # Start the download in a separate thread if not already downloading
-        if film_id not in download_threads or not download_threads[film_id].is_alive():
-            thread = threading.Thread(target=download_file, args=(file_url, TOKEN, cache_path, proxies, film_id))
-            download_threads[film_id] = thread
-            thread.start()
-        
-        return jsonify({"status": "Download started", "film_id": film_id})
+    file_url = f"https://huggingface.co/Unicone-Studio/jellyfin_media/resolve/main/{movie_path}"
+    proxies = get_system_proxies()
+    film_id = get_film_id(title)
     
-    return send_from_directory(os.path.dirname(cache_path), os.path.basename(cache_path))
+    # Start the download in a separate thread if not already downloading
+    if film_id not in download_threads or not download_threads[film_id].is_alive():
+        thread = threading.Thread(target=download_file, args=(file_url, TOKEN, cache_path, proxies, film_id, title))
+        download_threads[film_id] = thread
+        thread.start()
+    
+    return jsonify({"status": "Download started", "film_id": film_id})
 
 @app.route('/api/progress/<film_id>', methods=['GET'])
 def get_progress(film_id):
@@ -131,6 +144,16 @@ def get_film_id_by_title():
     film_id = get_film_id(title)
     return jsonify({"film_id": film_id})
 
+@app.route('/api/film_store', methods=['GET'])
+def get_film_store():
+    """Endpoint to get the film store JSON."""
+    if os.path.exists(FILM_STORE_JSON_PATH):
+        with open(FILM_STORE_JSON_PATH, 'r') as json_file:
+            film_store_data = json.load(json_file)
+        return jsonify(film_store_data)
+    return jsonify({}), 404
+
+
 # Main entry point
 if __name__ == "__main__":
     app.run(debug=True, host="0.0.0.0", port=7860)

hf_scrapper.py

@@ -28,7 +28,7 @@ def get_system_proxies():
         print(f"Error getting system proxies: {e}")
         return {}
 
-def download_file(file_url, token, cache_path, proxies, film_id, chunk_size=100 * 1024 * 1024):  # 100MB chunk size
+def download_file(file_url, token, cache_path, proxies, film_id, title, chunk_size=100 * 1024 * 1024):
     """
     Downloads a file from the specified URL and saves it to the cache path.
     Tracks the download progress.
@@ -39,6 +39,7 @@ def download_file(file_url, token, cache_path, proxies, film_id, chunk_size=100
         cache_path (str): The path to save the downloaded file.
         proxies (dict): Proxies for the request.
         film_id (str): Unique identifier for the film download.
+        title (str): The title of the film.
         chunk_size (int): Size of each chunk to download.
     """
     print(f"Downloading file from URL: {file_url} to {cache_path} with proxies: {proxies}")
@@ -58,7 +59,7 @@ def download_file(file_url, token, cache_path, proxies, film_id, chunk_size=100
                 download_progress[film_id]["downloaded"] += len(data)
         
         print(f'File cached to {cache_path} successfully.')
-        update_cache_json(file_url, cache_path)
+        update_film_store_json(title, cache_path)
     except RequestException as e:
         print(f"Error downloading file: {e}")
     except IOError as e:
@@ -84,25 +85,27 @@ def get_download_progress(film_id):
         return {"total": total, "downloaded": downloaded, "progress": progress}
     return {"total": 0, "downloaded": 0, "progress": 0}
 
-def update_cache_json(file_url, cache_path):
+def update_film_store_json(title, cache_path):
     """
-    Updates the cached films JSON with the new file.
+    Updates the film store JSON with the new file.
 
     Args:
-        file_url (str): The URL of the downloaded file.
+        title (str): The title of the film.
         cache_path (str): The local path where the file is saved.
     """
-    cache_data = {}
-    if os.path.exists(CACHE_JSON_PATH):
-        with open(CACHE_JSON_PATH, 'r') as json_file:
-            cache_data = json.load(json_file)
+    FILM_STORE_JSON_PATH = os.path.join(CACHE_DIR, "film_store.json")
 
-    film_title = os.path.basename(cache_path)
-    cache_data[film_title] = cache_path
+    film_store_data = {}
+    if os.path.exists(FILM_STORE_JSON_PATH):
+        with open(FILM_STORE_JSON_PATH, 'r') as json_file:
+            film_store_data = json.load(json_file)
+
+    film_store_data[title] = cache_path
+
+    with open(FILM_STORE_JSON_PATH, 'w') as json_file:
+        json.dump(film_store_data, json_file, indent=2)
+    print(f'Film store updated with {title}.')
 
-    with open(CACHE_JSON_PATH, 'w') as json_file:
-        json.dump(cache_data, json_file, indent=2)
-    print(f'Cache updated with {film_title}.')
 
 def get_file_structure(repo, token, path="", proxies=None):
     """