Update app.py

app.py

@@ -29,6 +29,29 @@ audio_model_name = "ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition"
 audio_processor = AutoFeatureExtractor.from_pretrained(audio_model_name)
 audio_model = AutoModelForAudioClassification.from_pretrained(audio_model_name).to(device)
 audio_sampling_rate = 16000
+import os
+
+def analyze_video(video_path):
+    if not video_path:
+        return "Error: No video input received!"
+    
+    # Check file size (limit to ~50MB for Hugging Face Spaces)
+    max_size_mb = 50
+    file_size_mb = os.path.getsize(video_path) / (1024 * 1024)
+
+    if file_size_mb > max_size_mb:
+        return f"Error: File size ({file_size_mb:.2f}MB) exceeds the {max_size_mb}MB limit."
+
+    # Process the video normally
+    audio_path = extract_audio(video_path)
+    frames = extract_frames(video_path)
+    facial_emotions = analyze_facial_emotion(frames)
+    transcription = transcribe_audio(audio_path)
+    audio_emotion, _ = analyze_audio_emotion(transcription)
+    final_emotion = max(facial_emotions, key=facial_emotions.get) if facial_emotions else "Neutral"
+
+    return transcription, audio_emotion, final_emotion, facial_emotions, "emotion_pie_chart.png"
+
 
 def calculate_angle(a, b, c):
     """Calculates the angle between three points."""