Appealing frontend

Author: aspects19

app/config.py

@@ -0,0 +1,7 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+class Config:
+    DB_URL = os.getenv("DB_URL")

app/database.py

@@ -1,13 +1,10 @@
 from sqlalchemy import create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
-from dotenv import load_dotenv
-import os
+from config import Config
 
-load_dotenv()
 
-DB_URL = os.getenv("DB_URL")
-engine = create_engine(DB_URL,echo=True)
+engine = create_engine(Config.DB_URL,echo=True)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 
 Base = declarative_base()

app/main.py

@@ -4,17 +4,17 @@
 import crud, models, schemas
 from pydantic import BaseModel
 from database import engine, get_db
-from fastapi.responses import FileResponse, JSONResponse
+from fastapi.responses import FileResponse, JSONResponse, HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from fastapi.middleware.gzip import GZipMiddleware
 from modules.downloader import download_video, download_audio
 
 app = FastAPI()
 
-# app.add_middleware(GZipMiddleware, minimum_size=1000, compresslevel=6)
+app.add_middleware(GZipMiddleware, minimum_size=1000, compresslevel=6)
 
-# app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
 templates = Jinja2Templates(directory="templates")
 
@@ -23,21 +23,21 @@
 
 class VideoRequest(BaseModel):
     url: str
+    resolution: str
+    format_type: str
 
 
-@app.get("/")
+@app.get("/", response_class=HTMLResponse)
 async def home(request: Request):
-    return templates.TemplateResponse(name="index.html", request=request)
+    return templates.TemplateResponse("index.html", {"request": request})
 
 
 @app.post("/download")
 async def download(request: VideoRequest):
     """ Download a video or audio based on the frontend request """
-    url = request.url
-    print(request)
 
     try:
-        video = await download_video(url)
+        video = await download_video(request.url, request.resolution)
         print(video)
         return FileResponse(path=f"{video}", media_type="video/mp4" )
     except Exception as e:

app/modules/downloader.py

@@ -59,7 +59,7 @@ def delete_file(f_path):
     except:
         time.sleep(5)
 
-def download_audio(video_url, output_path, thumb):
+def download_audio(video_url, output_path,):
     """Downloads audio with metadata and thumbnail."""
 
     try:
@@ -85,14 +85,15 @@ def download_audio(video_url, output_path, thumb):
     except Exception as e:
         print(f"Audio download failed: {e}")
 
-async def download_video(url ):
+async def download_video(url, resolution ):
     """Downloads video with user-selected format."""
     output_path="downloads/%(title)s.%(ext)s"
     ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
+            'format': f'bestvideo[height={resolution}]+bestaudio/best',
             'merge_output_format': 'mp4',
             'outtmpl': output_path
         }
+    print(resolution)
     try:
 
         # if domain == "instagram.com":
@@ -104,7 +105,8 @@ async def download_video(url ):
             video_info = ydl.extract_info(url, download=True)
             video_file = ydl.prepare_filename(video_info)
 
-        print(video_file)
+        # video_file = "/downloads/1 Second Video.mp4"
+        
         return str(video_file)
 
     except Exception as e:

static/bootstrap.min.css

@@ -0,0 +1,3 @@
+/* body {
+    background-color: #0f051f;
+} */