Update app.py

app.py

@@ -33,22 +33,24 @@ def krypton(input, history):
     print(f"Input: {input}")  # Debug input
     print(f"History: {history}")  # Debug history
 
+    image_path = None
     if input["files"]:
-        print("Found the image\n")
+        print("Found the image")
         image_path = input["files"][-1]["path"] if isinstance(input["files"][-1], dict) else input["files"][-1]
         print(f"Image path: {image_path}")
     else:
-        image_path = None
         for hist in history:
             if isinstance(hist[0], tuple):
                 image_path = hist[0][0]
-    
+                break
+
     if not image_path:
         gr.Error("You need to upload an image for Krypton to work.")
         return
 
     try:
         image = Image.open(image_path)
+        image.show()  # Show the image to confirm it's loaded
         print(f"Image open: {image}")
     except Exception as e:
         print(f"Error opening image: {e}")