Comfy

.gitignore

@@ -1 +1,2 @@
-/venv
+/venv
+.DS_Store

app.py

@@ -26,6 +26,9 @@ def extract_metadata(image_file):
     negative_prompt = 'N/A'
     adetailer_prompt_1 = 'N/A'
     adetailer_prompt_2 = 'N/A'
+    seed_number = -1
+    comfy_prompt = 'N/A'
+    comfy_workflow = 'N/A'
 
     # Convert the entire metadata dictionary to a readable string
     metadata_str = "\n".join([f"{key}: {value}" for key, value in metadata.items()])
@@ -64,22 +67,41 @@ def extract_metadata(image_file):
         if seed_match:
             seed_number = seed_match.group(1).strip()
 
+    # Extract additional metadata fields directly if they exist
+    if 'prompt' in metadata:
+        comfy_prompt = metadata['prompt'].strip()
+
+    if 'workflow' in metadata:
+        comfy_workflow = metadata['workflow'].strip()
+
     else:
         # No 'parameters' field in metadata
         prompt = 'N/A'
 
     # Return the outputs, replacing 'original_metadata_output' with 'metadata_str'
-    return img_display, prompt, negative_prompt, seed_number, adetailer_prompt_1, adetailer_prompt_2, metadata_str
+    return (
+        img_display, 
+        prompt, 
+        negative_prompt, 
+        seed_number, 
+        adetailer_prompt_1, 
+        adetailer_prompt_2, 
+        metadata_str, 
+        comfy_prompt, 
+        comfy_workflow
+        )
+
+
 
 # Create the Gradio interface
 with gr.Blocks() as iface:
     gr.Markdown("# Image Metadata Extractor")
     gr.Markdown("Upload an image (PNG, JPEG, TIFF) to extract its metadata and parse it for prompts.")
-
     with gr.Row():
         with gr.Column():  # First column for image upload and preview
             image_input = gr.File(label="Upload Image", type="filepath")
             image_output = gr.Image(label="Image Preview")
+            original_metadata_output = gr.Textbox(label="Original Metadata", lines=20)
 
         with gr.Column():  # Second column for metadata outputs
             prompt_output = gr.Textbox(label="Prompt", lines=4, show_copy_button=True)
@@ -87,8 +109,11 @@ with gr.Blocks() as iface:
             seed_output = gr.Textbox(label="Seed Number", lines=1, show_copy_button=True)
             adetailer_prompt_1_output = gr.Textbox(label="ADetailer Prompt 1", lines=3, show_copy_button=True)
             adetailer_prompt_2_output = gr.Textbox(label="ADetailer Prompt 2", lines=3, show_copy_button=True)
-            with gr.Row():  # Second column for metadata outputs
-                original_metadata_output = gr.Textbox(label="Original Metadata", lines=10)
+        with gr.Column():  # Second column for metadata outputs
+            with gr.Row():
+                comfy_prompt_output = gr.Textbox(label="Comfy Prompt", lines=20)
+            with gr.Row():
+                comfy_workflow_output = gr.Textbox(label="Comfy Workflow", lines=20)
 
     # Set up the input-output relationships
     image_input.change(
@@ -102,6 +127,8 @@ with gr.Blocks() as iface:
             adetailer_prompt_1_output,
             adetailer_prompt_2_output,
             original_metadata_output,
+            comfy_prompt_output, 
+            comfy_workflow_output,
         ]
     )