Update app.py

app.py

@@ -110,18 +110,26 @@ def extract_model_info(workflow_data):
                 models.append(inputs.get("model_path", "Unknown"))
     return "\n".join(models) if models else "Not found"
 
-def extract_image_info(workflow_data):
-    width = height = batch_size = "Not found"
+def extract_image_info_from_file(image_path):
+    """Extract actual image dimensions from the image file itself"""
+    try:
+        with Image.open(image_path) as img:
+            width, height = img.size
+            return str(width), str(height)
+    except Exception as e:
+        return "Not found", "Not found"
+
+def extract_batch_size(workflow_data):
+    """Extract batch size from workflow data"""
+    batch_size = "Not found"
     if not isinstance(workflow_data, dict):
-        return width, height, batch_size
+        return batch_size
     for node in workflow_data.values():
         if isinstance(node, dict) and node.get("class_type", "") == "EmptyLatentImage":
             inputs = node.get("inputs", {})
-            width = inputs.get("width", "Not found")
-            height = inputs.get("height", "Not found")
             batch_size = inputs.get("batch_size", "Not found")
             break
-    return str(width), str(height), str(batch_size)
+    return str(batch_size)
 
 def extract_nodes_info(workflow_data):
     if not isinstance(workflow_data, dict):
@@ -155,7 +163,11 @@ def process_single_image(image_path):
         positive, negative = extract_prompts(workflow_data)
         loras = extract_loras(workflow_data)
         models = extract_model_info(workflow_data)
-        width, height, batch = extract_image_info(workflow_data)
+
+        # Get actual image dimensions instead of workflow dimensions
+        width, height = extract_image_info_from_file(image_path)
+        batch = extract_batch_size(workflow_data)
+
         nodes = extract_nodes_info(workflow_data)
         full_json = extract_workflow_as_json(workflow_data)
     else:
@@ -370,4 +382,4 @@ with gr.Blocks(title="ComfyUI Workflow Extractor", css=css, theme="Werli/Purple-
     gr.Markdown("---\n💡 **Note:** It's under development.")
 
 if __name__ == "__main__":
-    demo.queue(max_size=10).launch(show_api=False, show_error=True)
+    demo.queue(max_size=10).launch(show_api=False, show_error=True)