Added error display

app.py

@@ -210,27 +210,35 @@ def predict(video_file, video_model_name, audio_model_name, framework_name):
 
     delete_directory_path = "./temp/"
 
-    video_path, audio_path = separate_video_audio(video_file)
+    try:
+        video_path, audio_path = separate_video_audio(video_file)
+        
+        video_prediction = predict_video(video_path, video_model, image_processor)
     
-    video_prediction = predict_video(video_path, video_model, image_processor)
-
-    highest_video_emotion = max(video_prediction, key=video_prediction.get)
+        highest_video_emotion = max(video_prediction, key=video_prediction.get)
+        
+        audio_prediction = preprocess_and_predict_audio(audio_path, audio_model, audio_processor)
     
-    audio_prediction = preprocess_and_predict_audio(audio_path, audio_model, audio_processor)
-
-    highest_audio_emotion = max(audio_prediction, key=audio_prediction.get)
+        highest_audio_emotion = max(audio_prediction, key=audio_prediction.get)
+        
+        framework_function = decision_frameworks[framework_name]
+        consensus_label = framework_function(video_prediction, audio_prediction)
+    
+        delete_files_in_directory(delete_directory_path)
     
-    framework_function = decision_frameworks[framework_name]
-    consensus_label = framework_function(video_prediction, audio_prediction)
+        result = f"""
+        <h2>Predictions</h2>
+        <p><strong>Video Label:</strong> {highest_video_emotion}</p>
+        <p><strong>Audio Label:</strong> {highest_audio_emotion}</p>
+        <p><strong>Consensus Label:</strong> {consensus_label}</p>
+        """
     
-    delete_files_in_directory(delete_directory_path)
+    except ValueError as e:
+        result = f"""
+        <h2>Error</h2>
+        <p>{str(e)}</p>
+        """
     
-    result = f"""
-    <h2>Predictions</h2>
-    <p><strong>Video Label:</strong> {highest_video_emotion}</p>
-    <p><strong>Audio Label:</strong> {highest_audio_emotion}</p>
-    <p><strong>Consensus Label:</strong> {consensus_label}</p>
-    """
     return result
 
 inputs = [