Update app.py

app.py

@@ -36,19 +36,37 @@ def check_nsfw(json_data, profile):
     return True
 
 def get_prompts_from_image(image_id):
+    headers = {
+        "authority": "civitai.com",
+        "accept": "*/*",
+        "accept-language": "en-BR,en;q=0.9,pt-BR;q=0.8,pt;q=0.7,es-ES;q=0.6,es;q=0.5,de-LI;q=0.4,de;q=0.3,en-GB;q=0.2,en-US;q=0.1,sk;q=0.1",
+        "content-type": "application/json",
+        "cookie": f'{os.environ["COOKIE_INFO"]}',
+        "if-modified-since": "Tue, 22 Aug 2023 07:18:52 GMT",
+        "referer": f"https://civitai.com/user/{username}/models",
+        "sec-ch-ua": "\"Not.A/Brand\";v=\"8\", \"Chromium\";v=\"114\", \"Google Chrome\";v=\"114\"",
+        "sec-ch-ua-mobile": "?0",
+        "sec-ch-ua-platform": "macOS",
+        "sec-fetch-dest": "empty",
+        "sec-fetch-mode": "cors",
+        "sec-fetch-site": "same-origin",
+        "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"
+    }
+    url = f'https://civitai.com/api/trpc/image.getGenerationData?input={{"json":{{"id":{image_id}}}}}'
+    response = requests.get(url, headers=headers)
     url = f'https://civitai.com/images/{image_id}'
     response = requests.get(url)
-    soup = BeautifulSoup(response.text, 'html.parser')
-    print(soup)
-    content = soup.find_all(class_='mantine-Text-root text-sm mantine-1c2skr8')
-    print(content)
-    if(content):
-        if(len(content) > 1):
-            return content[0].text, content[1].text
-        else:
-            return content[0].text, ""
-    else:
-        return "", ""
+    prompt = ""
+    negative_prompt = ""
+    if response.status_code == 200:
+        data = response.json()
+        result = data['result']['data']['json']
+        if "prompt" in result['meta']:
+            prompt = result['meta']['prompt']
+        if "negativePrompt" in result['meta']:
+            negative_prompt = result["meta"]["negativePrompt"]
+    
+    return prompt, negative_prompt
 
 def extract_info(json_data):
     if json_data["type"] == "LORA":