Update app.py

app.py

@@ -6,10 +6,9 @@ import logging
 from moviepy.editor import VideoFileClip
 import torch
 
-# Retrieve the access key from environment variables (Hugging Face secret)
+
 ACCESS_KEY = os.getenv("ACCESS_KEY")
 
-# Configure PyTorch for deterministic behavior
 torch.set_num_threads(1)
 torch.set_num_interop_threads(1)
 torch.use_deterministic_algorithms(True)
@@ -19,7 +18,7 @@ torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cudnn.allow_tf32 = False
 
 def truncate_video(video_file):
-    """Truncates video to 15 seconds and saves it as a temporary file."""
+    
     clip = VideoFileClip(video_file)
     truncated_clip = clip.subclip(0, min(15, clip.duration))
     truncated_video_file = "temp_truncated_video.mp4"
@@ -27,7 +26,6 @@ def truncate_video(video_file):
     return truncated_video_file
 
 def clone_repo():
-    """Clone the GitHub repository containing the backend."""
     repo_url = "https://github.com/NeeravSood/AllMark-MVP.git"
     repo_path = "./repository"
 
@@ -53,7 +51,6 @@ def clone_repo():
             raise RuntimeError(f"Failed to clone repository: {e.stderr}")
 
 def import_backend_script(script_name):
-    """Dynamically import the backend script."""
     try:
         script_path = os.path.join("./repository", script_name)
         if not os.path.exists(script_path):
@@ -67,7 +64,7 @@ def import_backend_script(script_name):
         logging.error(f"Error importing backend script: {str(e)}")
         raise RuntimeError(f"Failed to import backend script: {str(e)}")
 
-# Run repository setup and model import
+
 clone_repo()
 backend = import_backend_script("app.py") 
 analyzer = backend.DeepfakeAnalyzer()
@@ -78,8 +75,7 @@ def analyze_video(video_file):
         logging.error("Access key not set in environment variables.")
         return {"error": "Server misconfiguration. Access key not set."}
 
-    # Assuming you want to enforce a predefined, hardcoded key internally
-    expected_key = "expected_internal_key_here"
+    expected_key = ACCESS_KEY  # Directly using the environment variable
     if ACCESS_KEY != expected_key:
         logging.error("Unauthorized access attempt.")
         return {"error": "Unauthorized access. Invalid key provided."}
@@ -88,7 +84,6 @@ def analyze_video(video_file):
         truncated_video = truncate_video(video_file)
         results = analyzer.analyze_media(truncated_video)
 
-        # Get combined_assessment and handle non-numeric values
         combined_assessment = results.get('combined_assessment', 0)
         if isinstance(combined_assessment, str) and combined_assessment.lower() == "deepfake":
             analysis_result = "a deepfake"