Update app.py

app.py

@@ -4,11 +4,43 @@ from gradio_client import Client
 from PIL import Image
 import requests
 from io import BytesIO
+import re
+
+# 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'}
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()  # Raise exception for HTTP errors
+        
+        # Use regular expressions to find the endpoint URL in the script content
+        endpoint_match = re.search(r'root":"(https://[^"]+)', response.text)
+        if endpoint_match:
+            endpoint_url = endpoint_match.group(1)
+            return endpoint_url
+        else:
+            return None
+    except requests.RequestException as e:
+        print("Error fetching URL:", e)
+        return None
+    except Exception as e:
+        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
-client_playground = Client("https://playgroundai-playground-v2-5.hf.space/--replicas/tjkvn/")
+# Initialize the Playground AI client with the fetched endpoint
+client_playground = Client(endpoint)
 
 def format_prompt(logo_request):
     system_prompt = """
@@ -55,7 +87,7 @@ def generate_image(prompt, width=1024, height=1024, guidance_scale=7.5):
 
     # Extract the image URL from the result
     image_path = result[0][0]["image"]
-    image_url = "https://playgroundai-playground-v2-5.hf.space/--replicas/tjkvn/file=" + image_path
+    image_url = endpoint + "/file=" + image_path
     
     # Fetch and display the result image
     response = requests.get(image_url)
@@ -103,11 +135,11 @@ with gr.Blocks(css=css) as app:
     ### Meta Information
     **Project Title**:Magical AI Logo Generator
     
-    **Github**: [https://github.com/pacnimo/gpt-prompt-generator](https://github.com/pacnimo/)
+    **Github**: [https://github.com/pacnimo/](https://github.com/pacnimo/)
     
-    **Description**: Magical AI Logo Generator is Free and Easy to Use. Create a GPT Prompt Based on the Logo Request. 1 Click Logo Generator.
+    **Description**: Magical AI Logo Generator is Free and Easy to Use. Create a Website Logo with just 1 Click.
     
     **Footer**: © 2024 by [pacnimo](https://github.com/pacnimo/). All rights reserved.
     """)
 
-app.launch(debug=True)
+app.launch(debug=True)