Update app.py

app.py

@@ -23,11 +23,10 @@ def extract_parameters(image, show_full_metadata=False):
             try:
                 # Parse as JSON if possible
                 prompt_data = json.loads(raw_prompt)
-                # Look for the main prompt text (often in a node like '51')
+                # Look for the main prompt text
                 main_prompt = None
                 for node_id, node in prompt_data.items():
                     if "inputs" in node and "text" in node["inputs"]:
-                        # Prioritize nodes with meaningful text input
                         text = node["inputs"]["text"]
                         if isinstance(text, str) and text.strip():
                             main_prompt = text
@@ -37,7 +36,6 @@ def extract_parameters(image, show_full_metadata=False):
                     main_prompt = "No clear prompt found in workflow."
 
                 if show_full_metadata:
-                    # Format full metadata with spacing
                     full_output = "Full Workflow Metadata:\n\n"
                     for node_id, node in prompt_data.items():
                         full_output += f"Node {node_id}:\n"
@@ -46,15 +44,12 @@ def extract_parameters(image, show_full_metadata=False):
                         full_output += "\n"
                     return f"Main Prompt:\n{main_prompt}\n\n{full_output}"
                 else:
-                    # Just the cleaned-up main prompt
                     return f"Main Prompt:\n{main_prompt}"
             except json.JSONDecodeError:
-                # If not JSON, treat as raw text
                 if show_full_metadata:
                     return f"Raw Prompt:\n{raw_prompt}\n\nFull Metadata:\n{metadata}"
                 return f"Main Prompt:\n{raw_prompt}"
         else:
-            # No "prompt" key, show all metadata if toggled
             if show_full_metadata:
                 output = "Full Metadata:\n\n"
                 for key, value in metadata.items():
@@ -65,16 +60,23 @@ def extract_parameters(image, show_full_metadata=False):
     except Exception as e:
         return f"Error processing image: {str(e)}"
 
-# Define the Gradio interface
+# Define the Gradio interface with attribution
 interface = gr.Interface(
     fn=extract_parameters,
     inputs=[
         gr.Image(type="filepath", label="Upload a ComfyUI Image"),
-        gr.Checkbox(label="Show Full Metadata", value=False)  # Changed 'default' to 'value'
+        gr.Checkbox(label="Show Full Metadata", value=False)
     ],
     outputs=gr.Textbox(label="Extracted Parameters", lines=10),
     title="ComfyUI Image to Parameters",
-    description="Upload a ComfyUI image to extract its prompt. Toggle 'Show Full Metadata' for the complete workflow.",
+    description=(
+        "Upload a ComfyUI image to extract its prompt. Toggle 'Show Full Metadata' for the complete workflow.\n\n"
+        "This app is inspired by and adapted from ImagePromptViewer by LordKa-Berlin. "
+        "Original creator: LordKa-Berlin. "
+        "Licensed under CC BY 4.0 (https://creativecommons.org/licenses/by/4.0/). "
+        "Source: https://github.com/LordKa-Berlin/ImagePromptViewer. "
+        "Modified by QES for Gradio integration."
+    ),
     theme="default"
 )