Update app.py

app.py

@@ -1,6 +1,6 @@
-import io
 import random
 import time
+from io import BytesIO
 from urllib.parse import quote
 
 import gradio as gr
@@ -10,7 +10,7 @@ from PIL import Image
 MAX_SEED = 2 ** 31 - 1
 session = requests.Session()
 
-def _fetch_url(*args, retries=5, **kwargs):
+def _fetch(*args, retries=5, **kwargs):
     for i in range(retries + 1):
         try:
             response = session.get(*args, **kwargs)
@@ -24,13 +24,6 @@ def _fetch_url(*args, retries=5, **kwargs):
                 return None
     return None
 
-def add_to_gallery(image, gallery=None):
-    if gallery is None:
-        gallery = []
-    if image is not None:
-        gallery = gallery + [image]
-    return gallery
-
 def generate(
     prompt,
     seed=None,
@@ -41,14 +34,19 @@ def generate(
     if seed is None or randomize_seed:
         seed = random.randint(0, MAX_SEED)
     url = f"https://pollinations.ai/p/{quote(prompt)}?nologo=true&private=true"
-    params = dict(prompt=prompt, seed=seed, width=width, height=height)
-    image = _fetch_url(url, params=params)
+    params = dict(
+        prompt=prompt,
+        seed=seed,
+        width=width,
+        height=height
+    )
+    image = _fetch(url, params=params)
     if image:
-        return Image.open(io.BytesIO(image))
+        return Image.open(BytesIO(image))
     return None
 
 with gr.Blocks(
-    title="Text-to-image UI",
+    title="Text-to-Image",
     css="footer {display: none !important}",
     theme=gr.themes.Base(
         primary_hue="red",
@@ -56,7 +54,7 @@ with gr.Blocks(
         neutral_hue="neutral"
     )
 ) as app:
-    gr.Markdown("# Text-to-Image UI")
+    gr.Markdown("# Text-to-Image")
     with gr.Tab("🎨 Generate"):
         result = gr.Image(
             label="Image",
@@ -73,7 +71,7 @@ with gr.Blocks(
                 container=False
             )
             btn = gr.Button("Generate", variant="primary")
-
+            
     with gr.Tab("⚙️ Settings"):
         with gr.Row():
             seed = gr.Slider(
@@ -109,13 +107,20 @@ with gr.Blocks(
             interactive=False
         )
 
-    with gr.Accordion("ℹ️ About this UI", open=False):
+    with gr.Accordion("ℹ️ About", open=False):
         gr.Markdown(f"""
         * Created by [🍒 cherry-ghosts community](https://hf.co/cherry-ghosts)
         * Powered by [🐝 Pollinations](https://pollinations.ai) | [GitHub](https://github.com/pollinations/pollinations)
         * Running on [Gradio](https://www.gradio.app) v{gr.__version__}
         """)
 
+    def add_to_gallery(img, gallery):
+        if gallery is None:
+            gallery = []
+        if img is not None:
+            gallery = gallery + [img]
+        return gallery
+
     btn.click(
         generate,
         inputs=[prompt, seed, randomize_seed, width, height],