Update app.py

app.py

@@ -63,31 +63,26 @@ custom_css = """
     --text-primary: #ffffff;
     --text-secondary: #a0a0c0;
 }
-
 body {
     background: var(--dark-bg) !important;
     color: var(--text-primary) !important;
     font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif !important;
 }
-
 .gr-block {
     background: var(--dark-panel) !important;
     border-radius: 12px !important;
     border: 1px solid rgba(0, 243, 255, 0.2) !important;
     box-shadow: 0 0 15px rgba(0, 243, 255, 0.1) !important;
 }
-
 .gr-box {
     border-color: rgba(0, 243, 255, 0.3) !important;
     color: var(--text-primary) !important;
     background: rgba(10, 10, 26, 0.7) !important;
 }
-
 h1, h2, h3, h4, label, .gr-label {
     color: var(--text-primary) !important;
     text-shadow: 0 0 5px rgba(0, 243, 255, 0.5);
 }
-
 .gr-button {
     background: linear-gradient(45deg, var(--neon-primary), var(--neon-secondary)) !important;
     color: black !important;
@@ -102,17 +97,14 @@ h1, h2, h3, h4, label, .gr-label {
     width: 100% !important;
     margin: 10px 0 !important;
 }
-
 .gr-button:not(:disabled):hover {
     transform: translateY(-2px);
     box-shadow: 0 0 15px var(--neon-primary), 0 0 30px rgba(0, 243, 255, 0.5) !important;
 }
-
 .gr-button:disabled {
     background: #4b5563 !important;
     box-shadow: none !important;
 }
-
 .status-indicator {
     display: inline-block;
     width: 12px;
@@ -123,7 +115,6 @@ h1, h2, h3, h4, label, .gr-label {
     box-shadow: 0 0 0 0 rgba(0, 255, 135, 0.7);
     animation: pulse 2s infinite;
 }
-
 @keyframes pulse {
     0% {
         box-shadow: 0 0 0 0 rgba(0, 255, 135, 0.7);
@@ -135,14 +126,12 @@ h1, h2, h3, h4, label, .gr-label {
         box-shadow: 0 0 0 0 rgba(0, 255, 135, 0);
     }
 }
-
 .status-text {
     color: var(--text-secondary);
     font-size: 0.9rem;
     display: flex;
     align-items: center;
 }
-
 .header {
     display: flex;
     justify-content: space-between;
@@ -154,7 +143,6 @@ h1, h2, h3, h4, label, .gr-label {
     border: 1px solid rgba(0, 243, 255, 0.3);
     box-shadow: 0 0 20px rgba(0, 243, 255, 0.2);
 }
-
 .title-section h1 {
     margin-bottom: 0.25rem;
     font-weight: 800;
@@ -163,39 +151,32 @@ h1, h2, h3, h4, label, .gr-label {
     -webkit-text-fill-color: transparent;
     text-shadow: 0 0 10px rgba(0, 243, 255, 0.5);
 }
-
 .title-section p {
     color: var(--text-secondary);
     margin: 0;
 }
-
 .instructions {
     background: linear-gradient(90deg, rgba(0,243,255,0.05) 0%, rgba(255,0,255,0.05) 100%) !important;
     padding: 1.5rem !important;
     margin-bottom: 2rem !important;
     border: 1px solid rgba(0, 243, 255, 0.2) !important;
 }
-
 .instructions h3 {
     margin-top: 0;
     margin-bottom: 0.75rem;
     color: var(--neon-primary) !important;
 }
-
 .instructions ul {
     margin-bottom: 0;
     padding-left: 1.5rem;
 }
-
 .instructions li {
     margin-bottom: 0.5rem;
     color: var(--text-secondary);
 }
-
 .instructions li:last-child {
     margin-bottom: 0;
 }
-
 .footer {
     text-align: center;
     margin-top: 2rem;
@@ -204,7 +185,6 @@ h1, h2, h3, h4, label, .gr-label {
     color: var(--text-secondary);
     font-size: 0.875rem;
 }
-
 .image-container {
     border: 2px solid rgba(0, 243, 255, 0.3);
     border-radius: 12px;
@@ -213,7 +193,6 @@ h1, h2, h3, h4, label, .gr-label {
     margin-bottom: 1.5rem;
     box-shadow: 0 0 15px rgba(0, 243, 255, 0.1);
 }
-
 .image-container .gr-label {
     background: rgba(0, 243, 255, 0.1);
     padding: 8px 12px;
@@ -221,7 +200,6 @@ h1, h2, h3, h4, label, .gr-label {
     margin-bottom: 10px;
     display: inline-block;
 }
-
 .control-panel {
     background: linear-gradient(90deg, rgba(0,243,255,0.08) 0%, rgba(255,0,255,0.08) 100%) !important;
     padding: 1.5rem !important;
@@ -229,7 +207,6 @@ h1, h2, h3, h4, label, .gr-label {
     border-radius: 12px !important;
     margin-bottom: 1.5rem !important;
 }
-
 .status-panel {
     background: var(--darker-panel) !important;
     padding: 1.5rem !important;
@@ -237,7 +214,6 @@ h1, h2, h3, h4, label, .gr-label {
     border-radius: 12px !important;
     box-shadow: 0 0 15px rgba(0, 255, 135, 0.1) !important;
 }
-
 .status-header {
     display: flex;
     align-items: center;
@@ -245,22 +221,18 @@ h1, h2, h3, h4, label, .gr-label {
     padding-bottom: 0.5rem;
     border-bottom: 1px solid rgba(0, 255, 135, 0.2);
 }
-
 .status-content {
     min-height: 100px;
 }
-
 .output-highlight {
     border: 2px solid var(--neon-accent) !important;
     box-shadow: 0 0 20px rgba(0, 255, 135, 0.3) !important;
 }
-
 .upload-text {
     color: var(--text-secondary);
     text-align: center;
     padding: 20px;
 }
-
 .progress-bar {
     height: 6px;
     background: rgba(0, 243, 255, 0.2);
@@ -268,7 +240,6 @@ h1, h2, h3, h4, label, .gr-label {
     margin: 10px 0;
     overflow: hidden;
 }
-
 .progress-fill {
     height: 100%;
     background: linear-gradient(90deg, var(--neon-primary), var(--neon-accent));
@@ -276,7 +247,6 @@ h1, h2, h3, h4, label, .gr-label {
     width: 0%;
     transition: width 0.3s ease;
 }
-
 .control-item {
     margin-bottom: 1rem;
     padding: 1rem;
@@ -284,11 +254,9 @@ h1, h2, h3, h4, label, .gr-label {
     border-radius: 8px;
     border: 1px solid rgba(0, 243, 255, 0.1);
 }
-
 .control-item:last-child {
     margin-bottom: 0;
 }
-
 .download-btn {
     background: linear-gradient(45deg, var(--neon-accent), #00cc70) !important;
     margin-top: 15px !important;
@@ -415,6 +383,13 @@ with gr.Blocks(
                     visible=False,
                     elem_classes="download-btn"
                 )
+                
+                # Add a file component for downloading
+                download_file = gr.File(
+                    label="Download Result",
+                    visible=False,
+                    interactive=False
+                )
     
     # Instructions at the bottom
     with gr.Row():
@@ -440,51 +415,62 @@ with gr.Blocks(
     </div>
     """)
     
-    # Function to update status
-    def update_status(status):
-        return status
+    # Function to save the output image and return file path
+    def save_output_image(img):
+        if img is None:
+            return None, gr.update(visible=False)
+        
+        # Convert PIL Image to numpy array
+        img_array = np.array(img)
+        
+        # Save the image to a temporary file
+        output_path = "swapped_result.jpg"
+        Image.fromarray(img_array).save(output_path)
+        
+        return output_path, gr.update(visible=True, value=output_path)
     
     # Enhanced submit function with status updates
     def process_swap(source_file, target_file, doFaceEnhancer, many_faces_option):
         if source_file is None or target_file is None:
-            yield "❌ Please upload both source and target images", None, gr.update(visible=False)
+            yield "❌ Please upload both source and target images", None, gr.update(visible=False), None, gr.update(visible=False)
             return
         
-        yield "🟡 Processing: Analyzing faces and detecting features...", None, gr.update(visible=False)
+        yield "🟡 Processing: Analyzing faces and detecting features...", None, gr.update(visible=False), None, gr.update(visible=False)
         time.sleep(1)
         
-        yield "🟡 Processing: Extracting facial landmarks and matching features...", None, gr.update(visible=False)
+        yield "🟡 Processing: Extracting facial landmarks and matching features...", None, gr.update(visible=False), None, gr.update(visible=False)
         time.sleep(1)
         
-        yield "🟡 Processing: Swapping faces and applying transformations...", None, gr.update(visible=False)
+        yield "🟡 Processing: Swapping faces and applying transformations...", None, gr.update(visible=False), None, gr.update(visible=False)
         time.sleep(1)
         
         if doFaceEnhancer:
-            yield "🟡 Processing: Enhancing face quality and refining details...", None, gr.update(visible=False)
+            yield "🟡 Processing: Enhancing face quality and refining details...", None, gr.update(visible=False), None, gr.update(visible=False)
             time.sleep(1)
         
         result = swap_face_image(source_file, target_file, doFaceEnhancer)
         
         if result is None:
-            yield "❌ Error: Failed to process images. Please try again with different images.", None, gr.update(visible=False)
+            yield "❌ Error: Failed to process images. Please try again with different images.", None, gr.update(visible=False), None, gr.update(visible=False)
         else:
-            yield "✅ Processing complete! Result is ready for download.", result, gr.update(visible=True)
+            file_path, file_update = save_output_image(result)
+            yield "✅ Processing complete! Result is ready for download.", result, gr.update(visible=True), file_path, file_update
     
     # Connect the button with enhanced function
     submit.click(
         fn=process_swap,
         inputs=[source_image, target_image, face_enhancer, many_faces],
-        outputs=[info_text, output_image, download_btn]
+        outputs=[info_text, output_image, download_btn, download_file]
     )
     
-    # Download button functionality
-    def download_result():
-        return "💾 Download initiated! Check your downloads folder."
+    # Download button functionality - show the file download component
+    def show_download():
+        return gr.update(visible=True)
 
     download_btn.click(
-        fn=download_result,
+        fn=show_download,
         inputs=None,
-        outputs=info_text
+        outputs=download_file
     )
 
 if __name__ == "__main__":