Upload app.py

app.py

@@ -70,10 +70,12 @@ def send_security_alert(message: str, details: Optional[dict] = None):
         # Send email
         server.send_message(msg)
         server.quit()
-        print("تم إرسال التنبيه بنجاح")
+        print("✅ تم إرسال التنبيه بنجاح")
+        return True
         
     except Exception as e:
-        print(f"فشل في إرسال التنبيه الأمني: {str(e)}")
+        print(f"❌ فشل في إرسال التنبيه: {str(e)}")
+        return False
 
 class SecurityManager:
     def __init__(self):
@@ -216,24 +218,78 @@ def download_with_progress(url: str, fname: str):
             size = file.write(data)
             bar.update(size)
 
-# Initialize face analyzer
-app = FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
-app.prepare(ctx_id=-1, det_size=(640, 640))
-
-# Initialize face swapper
-def init_face_swapper():
-    model_path = os.path.expanduser('~/.insightface/models/inswapper_128.onnx')
-    model_url = "https://huggingface.co/ashleykleynhans/inswapper/resolve/main/inswapper_128.onnx"
-    
-    if not os.path.exists(model_path):
-        print("تحميل نموذج تبديل الوجوه...")
-        os.makedirs(os.path.dirname(model_path), exist_ok=True)
-        download_with_progress(model_url, model_path)
-    
-    return insightface.model_zoo.get_model(model_path, providers=['CPUExecutionProvider'])
-
-# Initialize swapper
-swapper = init_face_swapper()
+@protected_function
+def enhance_video(video_path, enhancement_type="all", progress=gr.Progress()):
+    """Enhance video quality"""
+    try:
+        if video_path is None:
+            return None, "الرجاء تحميل فيديو للتحسين"
+            
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Copy input video
+            input_path = os.path.join(temp_dir, "input.mp4")
+            shutil.copy2(video_path, input_path)
+            
+            # Prepare output path
+            output_path = os.path.join(temp_dir, "enhanced.mp4")
+            
+            # Get video properties
+            cap = cv2.VideoCapture(input_path)
+            total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+            fps = cap.get(cv2.CAP_PROP_FPS)
+            width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+            height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+            
+            # Prepare enhanced video writer
+            fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+            out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+            
+            frame_count = 0
+            progress(0, desc="جاري تحسين الفيديو...")
+            
+            while cap.isOpened():
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                    
+                # Apply enhancements based on type
+                if enhancement_type in ["all", "denoise"]:
+                    # Denoise frame
+                    frame = cv2.fastNlMeansDenoisingColored(frame, None, 10, 10, 7, 21)
+                    
+                if enhancement_type in ["all", "sharpen"]:
+                    # Sharpen frame
+                    kernel = np.array([[-1,-1,-1], [-1,9,-1], [-1,-1,-1]])
+                    frame = cv2.filter2D(frame, -1, kernel)
+                    
+                if enhancement_type in ["all", "contrast"]:
+                    # Enhance contrast
+                    lab = cv2.cvtColor(frame, cv2.COLOR_BGR2LAB)
+                    l, a, b = cv2.split(lab)
+                    clahe = cv2.createCLAHE(clipLimit=3.0, tileGridSize=(8,8))
+                    l = clahe.apply(l)
+                    lab = cv2.merge((l,a,b))
+                    frame = cv2.cvtColor(lab, cv2.COLOR_LAB2BGR)
+                
+                # Write enhanced frame
+                out.write(frame)
+                
+                # Update progress
+                frame_count += 1
+                progress(frame_count / total_frames)
+            
+            # Release resources
+            cap.release()
+            out.release()
+            
+            # Copy to final location
+            final_output = tempfile.mktemp(suffix='.mp4')
+            shutil.copy2(output_path, final_output)
+            
+            return final_output, "تم تحسين الفيديو بنجاح!"
+            
+    except Exception as e:
+        return None, f"حدث خطأ أثناء تحسين الفيديو: {str(e)}"
 
 @protected_function
 def process_frame(frame, source_face):
@@ -429,25 +485,67 @@ def send_test_alert():
     except Exception as e:
         return f"فشل في إرسال التنبيه التجريبي: {str(e)}"
 
+# Initialize face analyzer
+app = FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
+app.prepare(ctx_id=-1, det_size=(640, 640))
+
+# Initialize face swapper
+def init_face_swapper():
+    model_path = os.path.expanduser('~/.insightface/models/inswapper_128.onnx')
+    model_url = "https://huggingface.co/ashleykleynhans/inswapper/resolve/main/inswapper_128.onnx"
+    
+    if not os.path.exists(model_path):
+        print("تحميل نموذج تبديل الوجوه...")
+        os.makedirs(os.path.dirname(model_path), exist_ok=True)
+        download_with_progress(model_url, model_path)
+    
+    return insightface.model_zoo.get_model(model_path, providers=['CPUExecutionProvider'])
+
+# Initialize swapper
+swapper = init_face_swapper()
+
 # Create Gradio interface
 with gr.Blocks(title="Vudwos") as demo:
     if not _app.verify_integrity():
         raise RuntimeError("فشل التحقق من سلامة التطبيق")
         
-    gr.Markdown("""# Vudwos - تبديل الوجوه في الفيديو
-    قم بتحميل صورة المصدر والفيديو لتبديل الوجوه
+    gr.Markdown("""# Vudwos - معالجة الفيديو الذكية
+    قم بتحميل صورة المصدر والفيديو لتبديل الوجوه، أو قم بتحسين جودة الفيديو
     """)
     
-    with gr.Row():
-        with gr.Column():
-            source_img = gr.Image(label="صورة المصدر")
-            video_input = gr.Video(label="الفيديو")
-            submit_btn = gr.Button("بدء التحويل")
-            
-        with gr.Column():
-            video_output = gr.Video(label="الفيديو المحول")
-            status_text = gr.Textbox(label="الحالة", interactive=False)
-            
+    with gr.Tabs():
+        with gr.TabItem("🎭 تبديل الوجوه"):
+            with gr.Row():
+                with gr.Column():
+                    source_img = gr.Image(label="صورة المصدر")
+                    video_input = gr.Video(label="الفيديو")
+                    submit_btn = gr.Button("بدء التحويل")
+                    
+                with gr.Column():
+                    video_output = gr.Video(label="الفيديو المحول")
+                    status_text = gr.Textbox(label="الحالة", interactive=False)
+                    
+        with gr.TabItem("✨ تحسين الفيديو"):
+            with gr.Row():
+                with gr.Column():
+                    enhance_video_input = gr.Video(label="الفيديو المراد تحسينه")
+                    enhancement_type = gr.Radio(
+                        choices=["all", "denoise", "sharpen", "contrast"],
+                        value="all",
+                        label="نوع التحسين",
+                        info="""
+                        - all: كل التحسينات
+                        - denoise: إزالة التشويش
+                        - sharpen: زيادة الحدة
+                        - contrast: تحسين التباين
+                        """
+                    )
+                    enhance_btn = gr.Button("تحسين الفيديو")
+                    
+                with gr.Column():
+                    enhanced_video_output = gr.Video(label="الفيديو المحسن")
+                    enhance_status = gr.Textbox(label="الحالة", interactive=False)
+    
     with gr.Row():
         donate_btn = gr.Button("💝 تبرع للمشروع")
         verify_btn = gr.Button("✅ تحقق من التبرع")
@@ -460,6 +558,12 @@ with gr.Blocks(title="Vudwos") as demo:
         outputs=[video_output, status_text]
     )
     
+    enhance_btn.click(
+        fn=enhance_video,
+        inputs=[enhance_video_input, enhancement_type],
+        outputs=[enhanced_video_output, enhance_status]
+    )
+    
     donate_btn.click(
         fn=open_donation,
         inputs=[],