Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,6 @@ import logging
|
|
6 |
from moviepy.editor import VideoFileClip
|
7 |
import torch
|
8 |
|
9 |
-
|
10 |
ACCESS_KEY = os.getenv("ACCESS_KEY")
|
11 |
|
12 |
torch.set_num_threads(1)
|
@@ -18,7 +17,6 @@ torch.backends.cuda.matmul.allow_tf32 = False
|
|
18 |
torch.backends.cudnn.allow_tf32 = False
|
19 |
|
20 |
def truncate_video(video_file):
|
21 |
-
|
22 |
clip = VideoFileClip(video_file)
|
23 |
truncated_clip = clip.subclip(0, min(15, clip.duration))
|
24 |
truncated_video_file = "temp_truncated_video.mp4"
|
@@ -51,6 +49,7 @@ def clone_repo():
|
|
51 |
raise RuntimeError(f"Failed to clone repository: {e.stderr}")
|
52 |
|
53 |
def import_backend_script(script_name):
|
|
|
54 |
try:
|
55 |
script_path = os.path.join("./repository", script_name)
|
56 |
if not os.path.exists(script_path):
|
@@ -64,26 +63,21 @@ def import_backend_script(script_name):
|
|
64 |
logging.error(f"Error importing backend script: {str(e)}")
|
65 |
raise RuntimeError(f"Failed to import backend script: {str(e)}")
|
66 |
|
67 |
-
|
68 |
clone_repo()
|
69 |
backend = import_backend_script("app.py")
|
70 |
analyzer = backend.DeepfakeAnalyzer()
|
71 |
|
72 |
def analyze_video(video_file):
|
73 |
-
# Validate the access key at runtime
|
74 |
if ACCESS_KEY is None:
|
75 |
logging.error("Access key not set in environment variables.")
|
76 |
return {"error": "Server misconfiguration. Access key not set."}
|
77 |
|
78 |
-
expected_key = ACCESS_KEY # Directly using the environment variable
|
79 |
-
if ACCESS_KEY != expected_key:
|
80 |
-
logging.error("Unauthorized access attempt.")
|
81 |
-
return {"error": "Unauthorized access. Invalid key provided."}
|
82 |
-
|
83 |
try:
|
84 |
truncated_video = truncate_video(video_file)
|
85 |
results = analyzer.analyze_media(truncated_video)
|
86 |
|
|
|
87 |
combined_assessment = results.get('combined_assessment', 0)
|
88 |
if isinstance(combined_assessment, str) and combined_assessment.lower() == "deepfake":
|
89 |
analysis_result = "a deepfake"
|
@@ -111,4 +105,4 @@ interface = gr.Interface(
|
|
111 |
)
|
112 |
|
113 |
if __name__ == "__main__":
|
114 |
-
interface.launch()
|
|
|
6 |
from moviepy.editor import VideoFileClip
|
7 |
import torch
|
8 |
|
|
|
9 |
ACCESS_KEY = os.getenv("ACCESS_KEY")
|
10 |
|
11 |
torch.set_num_threads(1)
|
|
|
17 |
torch.backends.cudnn.allow_tf32 = False
|
18 |
|
19 |
def truncate_video(video_file):
|
|
|
20 |
clip = VideoFileClip(video_file)
|
21 |
truncated_clip = clip.subclip(0, min(15, clip.duration))
|
22 |
truncated_video_file = "temp_truncated_video.mp4"
|
|
|
49 |
raise RuntimeError(f"Failed to clone repository: {e.stderr}")
|
50 |
|
51 |
def import_backend_script(script_name):
|
52 |
+
"""Dynamically import the backend script."""
|
53 |
try:
|
54 |
script_path = os.path.join("./repository", script_name)
|
55 |
if not os.path.exists(script_path):
|
|
|
63 |
logging.error(f"Error importing backend script: {str(e)}")
|
64 |
raise RuntimeError(f"Failed to import backend script: {str(e)}")
|
65 |
|
66 |
+
# Run repository setup and model import
|
67 |
clone_repo()
|
68 |
backend = import_backend_script("app.py")
|
69 |
analyzer = backend.DeepfakeAnalyzer()
|
70 |
|
71 |
def analyze_video(video_file):
|
|
|
72 |
if ACCESS_KEY is None:
|
73 |
logging.error("Access key not set in environment variables.")
|
74 |
return {"error": "Server misconfiguration. Access key not set."}
|
75 |
|
|
|
|
|
|
|
|
|
|
|
76 |
try:
|
77 |
truncated_video = truncate_video(video_file)
|
78 |
results = analyzer.analyze_media(truncated_video)
|
79 |
|
80 |
+
|
81 |
combined_assessment = results.get('combined_assessment', 0)
|
82 |
if isinstance(combined_assessment, str) and combined_assessment.lower() == "deepfake":
|
83 |
analysis_result = "a deepfake"
|
|
|
105 |
)
|
106 |
|
107 |
if __name__ == "__main__":
|
108 |
+
interface.launch()
|