Update app.py

app.py

@@ -42,7 +42,10 @@ def get_prompts_from_image(image_id):
     soup = BeautifulSoup(response.text, 'html.parser')
     content = soup.find_all(class_='mantine-Code-root mantine-Code-block mantine-2v44jn')
     if(content):
-        return content[0].text
+        if(len(content) > 1):
+            return content[0].text, content[1].text
+        else:
+            return content[0].text, ""
     else:
         return ""
 
@@ -58,7 +61,7 @@ def extract_info(json_data):
                         # Then append all image URLs to the list
                         for image in model_version["images"]:
                             image_id = image["url"].split("/")[-1]
-                            prompt = get_prompts_from_image(image_id)
+                            prompt, negative_prompt = get_prompts_from_image(image_id)
                             if image["nsfwLevel"] > 2:
                                 pass #ugly before checking the actual logic
                             else:
@@ -66,7 +69,8 @@ def extract_info(json_data):
                                     "url": image["url"],
                                     "filename": os.path.basename(image["url"]),
                                     "type": "imageName",
-                                    "prompt": prompt #if "meta" in image and "prompt" in image["meta"] else ""
+                                    "prompt": prompt, #if "meta" in image and "prompt" in image["meta"] else ""
+                                    "negative_prompt": negative_prompt
                                 })
                         model_mapping = {
                             "SDXL 1.0": "stabilityai/stable-diffusion-xl-base-1.0",
@@ -102,6 +106,7 @@ def download_files(info, folder="."):
     downloaded_files = {
         "imageName": [],
         "imagePrompt": [],
+        "imageNegativePrompt": [],
         "weightName": []
     }
     for item in info["urls_to_download"]:
@@ -109,7 +114,9 @@ def download_files(info, folder="."):
         downloaded_files[item["type"]].append(item["filename"])
         if(item["type"] == "imageName"):
             prompt_clean = re.sub(r'<.*?>', '', item["prompt"])
+            negative_prompt_clean = re.sub(r'<.*?>', '', item["negative_prompt"])
             downloaded_files["imagePrompt"].append(prompt_clean)
+            downloaded_files["imageNegativePrompt"].append(negative_prompt_clean)
     return downloaded_files
 
 def download_file(url, filename, folder="."):
@@ -159,9 +166,11 @@ You should use {formatted_words} to trigger the image generation.
         trigger_words_section = ""
     
     widget_content = ""
-    for index, (prompt, image) in enumerate(zip(downloaded_files["imagePrompt"], downloaded_files["imageName"])):
+    for index, (prompt, negative_prompt, image) in enumerate(zip(downloaded_files["imagePrompt"], downloaded_files["imageNegativePrompt"], downloaded_files["imageName"])):
         escaped_prompt = prompt.replace("'", "''")
         widget_content += f"""- text: '{escaped_prompt if escaped_prompt else ' ' }'
+  parameters:
+      negative_prompt: {negative_prompt}
   output:
     url: >-
       {image}