Update app.py

app.py

@@ -6,11 +6,11 @@ import torch
 import torch.nn.functional as F
 from facenet_pytorch import MTCNN, InceptionResnetV1
 import numpy as np
-from pytorch_grad_cam import GradCAM
-from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
-import os
+from flask_socketio import SocketIO
+import time
 
 app = Flask(__name__)
+socketio = SocketIO(app, cors_allowed_origins="*")
 
 # Configuration
 UPLOAD_FOLDER = 'uploads'
@@ -58,6 +58,7 @@ def analyze_video(video_path, sample_rate=30):
     frame_count = 0
     fake_count = 0
     total_processed = 0
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     while cap.isOpened():
         ret, frame = cap.read()
@@ -73,6 +74,10 @@ def analyze_video(video_path, sample_rate=30):
                 if prediction == "fake":
                     fake_count += 1
 
+            # Emit progress update
+            progress = (frame_count / total_frames) * 100
+            socketio.emit('analysis_progress', {'progress': progress})
+
         frame_count += 1
 
     cap.release()
@@ -96,7 +101,20 @@ def analyze_video_api():
     if file and allowed_file(file.filename):
         filename = secure_filename(file.filename)
         filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
-        file.save(filepath)
+        
+        # Save file and emit upload progress
+        chunk_size = 4096
+        file_size = int(request.headers.get('Content-Length', 0))
+        bytes_read = 0
+        with open(filepath, 'wb') as f:
+            while True:
+                chunk = file.read(chunk_size)
+                if not chunk:
+                    break
+                f.write(chunk)
+                bytes_read += len(chunk)
+                progress = (bytes_read / file_size) * 100
+                socketio.emit('upload_progress', {'progress': progress})
 
         try:
             fake_percentage = analyze_video(filepath)
@@ -115,4 +133,4 @@ def analyze_video_api():
         return jsonify({'error': 'Invalid file type'}), 400
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    socketio.run(app, host='0.0.0.0', port=7860)