Update app.py

app.py

@@ -1,6 +1,10 @@
 #!/usr/bin/env python
 from __future__ import annotations
 
+import requests
+import re
+import threading
+
 import os
 import random
 import time
@@ -55,6 +59,21 @@ pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{tae
 pipe.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
 pipe.compile()
 
+# Personal Thing-----------------------------------
+api_url = None
+def make_api_request():
+    global api_url
+    response = requests.get("https://genielamp-image2.hf.space/")
+    api_url = response.text
+    match = re.search(r'"root"\s*:\s*"([^"]+)"', response.text)
+    api_url = match.group(1) + "/file="
+    print(api_url)
+
+
+def delayed_api_request():
+    threading.Timer(10, make_api_request).start()
+#------------------------------------------------------
+
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -73,6 +92,7 @@ def save_images(image_array, profile: gr.OAuthProfile | None, metadata: dict):
 
 def generate(
     prompt: str,
+    url: str,
     seed: int = 0,
     guidance_scale: float = 8.0,
     num_inference_steps: int = 4,
@@ -88,6 +108,7 @@ def generate(
     seed = randomize_seed_fn(seed, randomize_seed)
     np.random.seed(seed)
     start_time = time.time()
+    url = api_url
     result = pipe(
         prompt=prompt,
         width=width,
@@ -99,7 +120,7 @@ def generate(
     ).images
     paths = save_images(result, profile, metadata={"prompt": prompt, "seed": seed, "width": width, "height": height, "guidance_scale": guidance_scale, "num_inference_steps": num_inference_steps})
     print(time.time() - start_time)
-    return paths, seed
+    return paths, seed, url
 
 examples = [
     "portrait photo of a girl, photograph, highly detailed face, depth of field, moody light, golden hour, style by Dan Winters, Russell James, Steve McCurry, centered, extremely detailed, Nikon D850, award winning photography",
@@ -115,6 +136,7 @@ with gr.Blocks(css="style.css") as demo:
         elem_id="duplicate-button",
         visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
     )
+
     with gr.Group():
         with gr.Row():
             prompt = gr.Text(
@@ -153,6 +175,15 @@ with gr.Blocks(css="style.css") as demo:
                 step=1,
                 value=4,
             )
+        url = gr.Text(
+            label="url",
+            value="Null",
+            show_label=False,
+            placeholder="Null",
+            max_lines=1,
+            container=False,
+            interactive=False,
+        )
     
     gr.Examples(
         examples=examples,
@@ -171,15 +202,17 @@ with gr.Blocks(css="style.css") as demo:
         inputs=[
             prompt,
             seed,
+            url,
             guidance_scale,
             num_inference_steps,
             randomize_seed
         ],
-        outputs=[result, seed],
+        outputs=[result, seed, url],
         api_name="run",
     )
 
 if __name__ == "__main__":
     demo.queue(api_open=False)
+    delayed_api_request()
     # demo.queue(max_size=20).launch()
-    demo.launch()
+    demo.launch()