update with audio / video url and gpu version of torch

app.py

@@ -22,6 +22,7 @@ async def analyze_audio(request: AudioRequest):
     if not audio_url:
         raise HTTPException(status_code=400, detail="audio_url is required")
     
+    print(f"Analyzing audio... {audio_url}")
     analyzer = AudioAnalyzer(media_url=audio_url, media_type="audio")
     traits = analyzer.retrieve_traits()
     
@@ -34,6 +35,7 @@ async def analyze_video(request: VideoRequest):
     if not video_url:
         raise HTTPException(status_code=400, detail="video_url is required")
     
+    print(f"Analyzing video... {video_url}")
     analyzer = VideoAnalyzer(video_url=video_url)
     traits = analyzer.retrieve_traits()
     

audio_analyzer.py

@@ -16,6 +16,7 @@ TWILIO_ACCOUNT_SID = os.getenv("TWILIO_ACCOUNT_SID")
 TWILIO_AUTH_TOKEN = os.getenv("TWILIO_AUTH_TOKEN")
 client = OpenAI(api_key=OPENAI_API_KEY)
 
+
 class AudioAnalyzer:
     def __init__(self, media_path=None, media_url=None, media_type=None):
         self.personality_labels = [
@@ -23,7 +24,7 @@ class AudioAnalyzer:
             "Creative", "Analytical", "Dependable", "Impulsive", "Adaptable", "Meticulous", "Assertive", 
             "Agreeable", "Courageous", "Cautious", "Patient", "Ambitious", "Generous"
         ]
-        self.personality_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+        # self.personality_pipeline = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
         self.media_path = media_path
         self.media_url = media_url
         self.media_type = media_type
@@ -81,8 +82,8 @@ class AudioAnalyzer:
             for chunk in chunks:
                 processed_audio += chunk
 
-        # trim to 1 minute
-        processed_audio = processed_audio[:60000]
+        # trim to 2 minutes
+        processed_audio = processed_audio[:120000]
 
         # Export as mp3
         processed_audio.export(mp3_file, format="mp3", bitrate=bitrate)

requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
 fastapi==0.111.1
 imageio==2.34.1
 moviepy==1.0.3