Update app.py

app.py

@@ -4,10 +4,12 @@ import importlib.util
 import gradio as gr
 import logging
 from moviepy.editor import VideoFileClip
-import json
-import spaces
 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)
@@ -16,8 +18,6 @@ torch.backends.cudnn.benchmark = False
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cudnn.allow_tf32 = False
 
-ACCESS_KEY = os.getenv("ACCESS_KEY")
-
 def truncate_video(video_file):
     """Truncates video to 15 seconds and saves it as a temporary file."""
     clip = VideoFileClip(video_file)
@@ -28,7 +28,7 @@ def truncate_video(video_file):
 
 def clone_repo():
     """Clone the GitHub repository containing the backend."""
-    repo_url = "https://github.com/NeeravSood/AllMark-MVP.git"  # Update when changing
+    repo_url = "https://github.com/NeeravSood/AllMark-MVP.git"
     repo_path = "./repository"
 
     github_pat = os.getenv("GITHUB_PAT")
@@ -67,15 +67,22 @@ 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()
 
-@spaces.GPU(duration=1000)
 def analyze_video(video_file):
-    if access_key != ACCESS_KEY:
-        logging.error("Invalid access key provided.")
-        return {"error": "Unauthorized access. Invalid access key."}
+    # Validate the access key at runtime
+    if ACCESS_KEY is None:
+        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"
+    if ACCESS_KEY != expected_key:
+        logging.error("Unauthorized access attempt.")
+        return {"error": "Unauthorized access. Invalid key provided."}
 
     try:
         truncated_video = truncate_video(video_file)
@@ -102,7 +109,7 @@ def analyze_video(video_file):
 
 interface = gr.Interface(
     fn=analyze_video,
-    inputs=[gr.Video(label="Upload Video"), gr.Textbox(label="Access Key", type="password")],
+    inputs=gr.Video(label="Upload Video"),
     outputs="json",
     title="AllMark - Deepfake Analyzer",
     description="Upload a video to analyze. N.B. - Only mp4 files. Processing time 1-10 minutes. For any false negatives, please contact the publisher for verification. Incognito Mode Recommended"