Update main.py

main.py

@@ -1,12 +1,12 @@
 from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.staticfiles import StaticFiles
-import shutil
+from huggingface_hub import HfApi
 import os
+import shutil
 
 app = FastAPI()
 
-# Разрешаем запросы с любого адреса (чтобы твой HTML мог общаться с сервером)
+# Разрешаем твоему сайту подключаться
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -14,24 +14,44 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-UPLOAD_DIR = "uploads"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
+# НАСТРОЙКИ (EmeraldCreator — это твой ник)
+REPO_ID = "EmeraldCreator/BlueTok-Storage" 
+TOKEN = os.environ.get("HF_TOKEN")
+
+api = HfApi()
 
 @app.post("/upload")
 async def upload_video(file: UploadFile = File(...)):
-    # Сохраняем файл с оригинальным именем
-    file_path = os.path.join(UPLOAD_DIR, file.filename)
-    with open(file_path, "wb") as buffer:
+    temp_path = file.filename
+    with open(temp_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
-    return {"url": f"/{UPLOAD_DIR}/{file.filename}"}
+    
+    try:
+        # Загружаем в датасет
+        api.upload_file(
+            path_or_fileobj=temp_path,
+            path_in_repo=f"videos/{file.filename}",
+            repo_id=REPO_ID,
+            repo_type="dataset",
+            token=TOKEN
+        )
+        
+        # Ссылка на файл в датасете
+        raw_url = f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/videos/{file.filename}"
+        os.remove(temp_path)
+        return {"url": raw_url}
+    except Exception as e:
+        return {"error": str(e)}
 
 @app.get("/videos")
 async def get_videos():
-    # Получаем список всех видео в папке
-    if not os.path.exists(UPLOAD_DIR):
+    try:
+        # Список всех файлов в датасете
+        files = api.list_repo_files(repo_id=REPO_ID, repo_type="dataset", token=TOKEN)
+        video_urls = [
+            f"https://huggingface.co/datasets/{REPO_ID}/resolve/main/{f}" 
+            for f in files if f.startswith("videos/") and f.endswith(('.mp4', '.mov', '.avi'))
+        ]
+        return video_urls
+    except:
         return []
-    videos = [f"/{UPLOAD_DIR}/{v}" for v in os.listdir(UPLOAD_DIR) if v.endswith(('.mp4', '.mov', '.avi'))]
-    return videos
-
-# Раздаем папку uploads как статические файлы
-app.mount("/uploads", StaticFiles(directory=UPLOAD_DIR), name="uploads")