Update app.py

app.py

@@ -76,25 +76,36 @@ def analyze_video(video_file):
         return {"error": "Server misconfiguration. Access key not set."}
 
     try:
-        truncated_video = truncate_video(video_file)  # Truncate the video for faster analysis, if required.
+        truncated_video = truncate_video(video_file)  # Truncate the video for faster analysis
         results = analyzer.analyze_media(truncated_video)
 
-        # Retrieve the final combined assessment result from the analysis
-        combined_assessment = results.get('combined_assessment', "Genuine")
-
-        # Determine the analysis result based on the combined assessment
-        if combined_assessment.lower() == "deepfake":
-            analysis_result = "a deepfake"
+        # Extract frame count and video probability
+        frame_count = len(results['video_analysis']['frame_results'])
+        video_probability = results['video_analysis']['probability']
+        combined_assessment = results.get('combined_assessment', "Inconclusive")
+
+        # Reapply combined_results logic for front-end clarity
+        if frame_count < 300:
+            assessment = "Inconclusive due to lack of sufficient frames"
+        elif frame_count < 500:
+            threshold = 0.4
+            assessment = "Deepfake" if video_probability >= threshold else "Genuine"
         else:
-            analysis_result = "genuine/original"
+            threshold = 0.5
+            assessment = "Deepfake" if video_probability >= threshold else "Genuine"
+
+        # Construct output message
+        message = (
+            f"According to our analysis, the video you uploaded appears to be {assessment}. "
+            f"{frame_count} frames were analyzed in total."
+        )
 
-        # Construct output with the analysis result and detailed breakdown
+        # Return the final output
         output = {
-            "message": f"According to our analysis, the video you uploaded appears to be {analysis_result}. "
-                       f"{len(results['video_analysis']['frame_results'])} frames were analyzed in total.",
+            "message": message,
             "details": {
-                "video_probability": results['video_analysis']['probability'],  # Video probability score
-                "combined_assessment": combined_assessment  # Final assessment (Deepfake or Genuine)
+                "video_probability": video_probability,
+                "combined_assessment": assessment  # Final assessment based on reapplied logic
             }
         }
         return output
@@ -105,6 +116,7 @@ def analyze_video(video_file):
 
 
 
+
 interface = gr.Interface(
     fn=analyze_video,
     inputs=gr.Video(label="Upload Video"),