Update app.py

app.py

@@ -1,11 +1,3 @@
-"""
-Complete Industrial Warehouse Image Stitching Application
-Ready for Hugging Face Space Deployment with ZIP Upload
-
-Author: Your Name
-Date: November 2024
-License: MIT
-"""
 
 import gradio as gr
 import cv2
@@ -242,7 +234,8 @@ class WarehouseStitcher:
         buf.seek(0)
         
         # Return (image preview, logs, downloadable file)
-        return result_pil, "\n".join(logs), ("stitched_result.png", buf.getvalue())
+        # return result_pil, "\n".join(logs), ("stitched_result.png", buf.getvalue())
+        return result_pil, "\n".join(logs), buf.getvalue()
 
 
 class PoseGuidedWarehouseStitcher(WarehouseStitcher):
@@ -412,7 +405,8 @@ class PoseGuidedWarehouseStitcher(WarehouseStitcher):
         result_pil.save(buf, format='PNG', optimize=True)
         buf.seek(0)
         
-        return result_pil, "\n".join(logs), buf
+        # return result_pil, "\n".join(logs), buf
+        return result_pil, "\n".join(logs), buf.getvalue()
 
 
 def create_demo():
@@ -435,20 +429,21 @@ def create_demo():
                 file_paths.append(f[0])
             else:
                 file_paths.append(f)
-                
-                valid_paths = []
-                for path in file_paths:
-                    try:
-                        img = cv2.imread(path)
-                        if img is not None:
-                            valid_paths.append(path)
-                    except:
-                        pass
-                
-                if len(valid_paths) < 2:
-                    return None, f"❌ Only {len(valid_paths)} valid images found", None
-                
-                file_paths = valid_paths
+        
+        # Validate images
+        valid_paths = []
+        for path in file_paths:
+            try:
+                img = cv2.imread(path)
+                if img is not None:
+                    valid_paths.append(path)
+            except:
+                pass
+        
+        if len(valid_paths) < 2:
+            return None, f"❌ Only {len(valid_paths)} valid images found", None
+        
+        file_paths = valid_paths
         
         # Update configuration
         basic_stitcher.config['feature_extractor'] = feature_type