Update app.py

app.py

@@ -8,6 +8,7 @@ import re
 import uuid
 from typing import Optional
 import json
+from bs4 import BeautifulSoup
 
 TRUSTED_UPLOADERS = ["KappaNeuro", "CiroN2022", "multimodalart", "Norod78", "joachimsallstrom", "blink7630", "e-n-v-y", "DoctorDiffusion", "RalFinger"]
 
@@ -35,6 +36,16 @@ def check_nsfw(json_data, profile):
                 return False
     return True
 
+def get_prompts_from_image(image_id):
+    url = 'https://civitai.com/images/9476412'
+    response = requests.get(url)
+    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
+    else:
+        return ""
+
 def extract_info(json_data):
     if json_data["type"] == "LORA":
         for model_version in json_data["modelVersions"]:
@@ -46,7 +57,8 @@ def extract_info(json_data):
                         
                         # Then append all image URLs to the list
                         for image in model_version["images"]:
-                            print(image)
+                            image_id = image["url"].split("/")[-1]
+                            prompt = get_prompts_from_image(image_id)
                             if image["nsfwLevel"] > 2:
                                 pass #ugly before checking the actual logic
                             else:
@@ -54,7 +66,7 @@ def extract_info(json_data):
                                     "url": image["url"],
                                     "filename": os.path.basename(image["url"]),
                                     "type": "imageName",
-                                    "prompt": image["meta"]["prompt"] if "meta" in image and "prompt" in image["meta"] else ""
+                                    "prompt": prompt #if "meta" in image and "prompt" in image["meta"] else ""
                                 })
                         model_mapping = {
                             "SDXL 1.0": "stabilityai/stable-diffusion-xl-base-1.0",