switch to fastapi

app.py

@@ -1,38 +1,40 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 from audio_analyzer import AudioAnalyzer
 from video_analyzer import VideoAnalyzer
 
-app = Flask(__name__)
+app = FastAPI()
 
-@app.route('/', methods=['GET'])
-def hello_world():
-    return jsonify({"message": "Hello, World!"})
+class AudioRequest(BaseModel):
+    audio_url: str
 
-@app.route('/v1/analyze_audio', methods=['POST'])
-def analyze_audio():
-    data = request.get_json()
-    audio_url = data.get('audio_url')
+class VideoRequest(BaseModel):
+    video_url: str
+
+@app.get("/")
+async def hello_world():
+    return {"message": "Hello, World!"}
+
+@app.post("/v1/analyze_audio")
+async def analyze_audio(request: AudioRequest):
+    audio_url = request.audio_url
     
     if not audio_url:
-        return jsonify({"error": "audio_url is required"}), 400
+        raise HTTPException(status_code=400, detail="audio_url is required")
     
     analyzer = AudioAnalyzer(media_url=audio_url, media_type="audio")
     traits = analyzer.retrieve_traits()
     
-    return jsonify(traits)
+    return traits
 
-@app.route('/v1/analyze_video', methods=['POST'])
-def analyze_video():
-    data = request.get_json()
-    video_url = data.get('video_url')
+@app.post("/v1/analyze_video")
+async def analyze_video(request: VideoRequest):
+    video_url = request.video_url
     
     if not video_url:
-        return jsonify({"error": "video_url is required"}), 400
+        raise HTTPException(status_code=400, detail="video_url is required")
     
     analyzer = VideoAnalyzer(video_url=video_url)
     traits = analyzer.retrieve_traits()
     
-    return jsonify(traits)
-
-if __name__ == '__main__':
-    app.run(debug=True)
+    return traits

requirements.txt

@@ -1,4 +1,4 @@
-flask==3.0.3
+fastapi==0.111.1
 moviepy==1.0.3
 transformers==4.42.4
 openai==1.34.0