Update app.py

app.py

@@ -5,8 +5,8 @@ from PIL import Image
 import requests
 from io import BytesIO
 import re
-import pandas as pd
 
+# Function to fetch the API endpoint from the URL using regular expressions
 def fetch_api_endpoint(url):
     try:
         headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'}
@@ -15,7 +15,7 @@ def fetch_api_endpoint(url):
         
         # Use regular expressions to find the endpoint URL in the script content
         endpoint_match = re.search(r'root":"(https://[^"]+)', response.text)
-        if (endpoint_match):
+        if endpoint_match:
             endpoint_url = endpoint_match.group(1)
             return endpoint_url
         else:
@@ -27,15 +27,19 @@ def fetch_api_endpoint(url):
         print("An unexpected error occurred:", e)
         return None
 
+# Define the URL to fetch the API endpoint from
 url = "https://playgroundai-playground-v2-5.hf.space/"
 
+# Fetch the API endpoint
 endpoint = fetch_api_endpoint(url)
 
 if endpoint is None:
     print("Failed to fetch the API endpoint.")
     exit(1)
 
+# Initialize the HuggingFace Inference Client with the specified model
 client_mistral = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2")
+# Initialize the Playground AI client with the fetched endpoint
 client_playground = Client(endpoint)
 
 def format_prompt(logo_request):
@@ -102,20 +106,11 @@ css = """
 }
 """
 
-results = []
-
 def process_request(logo_request, width, height, guidance_scale):
     improved_prompt = generate_improved_prompt(logo_request)
     image = generate_image(improved_prompt, width, height, guidance_scale)
-    results.append({"Prompt": logo_request, "Guidance Scale": guidance_scale, "Size": f"{width}x{height}", "Image": image})
-    display_results()
     return improved_prompt, image
 
-def display_results():
-    df = pd.DataFrame(results)
-    html = df.to_html(escape=False, formatters=dict(Image=lambda x: x._repr_html_()))
-    display_html(html, raw=True)
-
 with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column(scale=2):
@@ -124,9 +119,6 @@ with gr.Blocks(css=css) as app:
             width = gr.Slider(label="Width", minimum=256, maximum=1536, value=1024)
             height = gr.Slider(label="Height", minimum=256, maximum=1536, value=1024)
             guidance_scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=20, value=3)
-            example1 = gr.Button("Example 1: Abstract geometric shape")
-            example2 = gr.Button("Example 2: Minimalist animal logo")
-            example3 = gr.Button("Example 3: Futuristic tech icon")
         with gr.Column(scale=3):
             gr.HTML("<h1><center>Magical AI Logo Generator<h1><center>")
             generate_button = gr.Button("Generate")
@@ -137,21 +129,6 @@ with gr.Blocks(css=css) as app:
                 inputs=[logo_input, width, height, guidance_scale],
                 outputs=[prompt_output, image_output]
             )
-            example1.click(
-                fn=lambda: ("Abstract geometric shape",),
-                inputs=None,
-                outputs=[logo_input]
-            )
-            example2.click(
-                fn=lambda: ("Minimalist animal logo",),
-                inputs=None,
-                outputs=[logo_input]
-            )
-            example3.click(
-                fn=lambda: ("Futuristic tech icon",),
-                inputs=None,
-                outputs=[logo_input]
-            )
 
     gr.Markdown("""
     ---