Update app.py

app.py

@@ -1,24 +1,26 @@
 from pytubefix import YouTube
 from pytubefix.cli import on_progress
-from flask import Flask
-from flask_cors import CORS
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
 
-app = Flask(__name__)
-CORS(app)
+app = FastAPI()
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Adjust as needed
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-@app.route('/')
+@app.get("/")
 def index():
-    return "Hello, World!"
+    return {"message": "Hello, World!"}
 
-
-@app.route('/api/video/<id>')
-def video_id(id):
+@app.get("/api/video/{id}")
+def video_id(id: str):
     url = f"https://www.youtube.com/watch?v={id}"
-
     yt = YouTube(url, on_progress_callback=on_progress)
-
     ys = yt.streams.get_highest_resolution()
-
     return {
         "title": yt.title,
         "url": ys.url,