Update app.py

app.py

@@ -159,23 +159,33 @@ def homeImageLoader():
 
 @app.route('/prompt/<id>', methods=["GET"])
 def promptDetail(id):
-    f = open('picxai.json', "r", encoding="utf-8")
-    data = json.loads(f.read())
-    flag = 0
-    for i in data["data"]:
-        for j in i["prompts"]:
-            if j["id"] == id:
-                datafinal = j
-                flag = 1
-                break
-    if flag == 0:
-        f = open('sd1-5.json', "r", encoding="utf-8")
-        data = json.loads(f.read())
-        for i in data["data"]:
-            for j in i["prompts"]:
-                if j["id"] == id:
-                    datafinal = j
-                    break
+    url = "https://lexica.art/prompt/" + id
+    payload={}
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
+    }
+    response = requests.request("GET", url, headers=headers, data=payload)
+    datafinal = {}
+    datafinal["id"] = id
+    datafinal["prompt"] = response.text.split('name="description" content="')[1].split('"')[0]
+    datafinal["c"] = int(response.text.split('Guidance scale')[1].split('<div class="text-sm">')[1].split('</div>')[0])
+    datafinal["width"] = int(response.text.split('Dimensions')[1].split('<div class="text-sm">')[1].split('</div>')[0].split("×")[0].replace("<!-- -->",""))
+    datafinal["height"] = int(response.text.split('Dimensions')[1].split('<div class="text-sm">')[1].split('</div>')[0].split("×")[0].replace("<!-- -->",""))
+    datafinal["images"] = []
+    tempImage = response.text.split('<meta property="og:image" content="')[1].split('"')[0].split("images=")[1:]
+    for i in tempImage:
+        datafinal["images"].append({"id": i.replace("&amp;", "").split("/")[-1]})
+    try:
+        datafinal["negativePrompt"] = response.text.split('Negative prompt')[1].split('<div class="text-sm">')[1].split('</div>')[0]
+    except:
+        datafinal["negativePrompt"] = ""
+    try:
+        datafinal["upscaled_width"] = int(response.text.split('Upscaled')[1].split('<div class="text-sm">')[1].split('</div>')[0].split("×")[0].replace("<!-- -->",""))
+        datafinal["upscaled_height"] = int(response.text.split('Upscaled')[1].split('<div class="text-sm">')[1].split('</div>')[0].split("×")[1].replace("<!-- -->",""))
+    except:
+        datafinal["upscaled_width"] =""
+        datafinal["upscaled_height"] =""
+    datafinal["model"] = "picxai-diffuser"
     return render_template("prompt.html", datafinal=datafinal)