Update app.py

app.py

@@ -21,42 +21,14 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
-    resolution = resolutions.get(resolution_key, (1024, 1024))
-    width, height = resolution
-    full_prompt = f"{prompt}, Canon EOS R5, 4K, Photo-Realistic, appearing photorealistic with super fine details, high resolution, natural look, hyper realistic photography, cinematic lighting, --ar 64:37, --v 6.0, --style raw, --stylize 750"
-    
-    try:
-        result = client_image.predict(
-            prompt=full_prompt,
-            negative_prompt="(deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers:1.4), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
-            use_negative_prompt=True,
-            style=style,
-            seed=0,
-            width=width,
-            height=height,
-            guidance_scale=5,
-            randomize_seed=True,
-            api_name="/run"
-        )
-        logger.info("Image generation successful.")
-        return result
-    except Exception as e:
-        logger.error(f"Error generating image: {e}")
-        return None        
-
-def gradio_interface(prompt, resolution_key):
-    result = generate_image(prompt, resolution_key)
-    
-    if result and result[0]:
-        file_path = result[0][0].get('image')
-        if file_path and os.path.exists(file_path):
-            return file_path, "The image was generated successfully."
-        else:
-            return None, "The image file is not available. Please try again later."
-    else:
-        return None, "There was an error processing your photo. Please try again later."
+# Helper function to extract username from URL
+def extract_username_from_url(url):
+    parsed_url = urlparse(url)
+    params = parse_qs(parsed_url.query)
+    logger.debug(f"Parsed URL parameters: {params}")  # Debugging information
+    return params.get('username', ["Guest"])[0]
 
+# Define the Gradio app
 def create_gradio_interface(username):
     with gr.Blocks() as interface:
         # Personalized HTML content
@@ -80,6 +52,19 @@ def create_gradio_interface(username):
         result_output = gr.Image(label="Generated Image", type="pil")
         message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
         
+        # Define the function for image generation
+        def gradio_interface(prompt, resolution_key):
+            result = generate_image(prompt, resolution_key)
+            
+            if result and result[0]:
+                file_path = result[0][0].get('image')
+                if file_path and os.path.exists(file_path):
+                    return file_path, "The image was generated successfully."
+                else:
+                    return None, "The image file is not available. Please try again later."
+            else:
+                return None, "There was an error processing your photo. Please try again later."
+        
         # Set up interaction
         generate_button.click(
             fn=lambda prompt, resolution_key: gradio_interface(prompt, resolution_key),
@@ -97,23 +82,13 @@ def create_gradio_interface(username):
     
     return interface
 
+# Launch Gradio app
 def launch_gradio(username):
     interface = create_gradio_interface(username)
     interface.launch()
 
-def extract_username_from_url(url):
-    parsed_url = urlparse(url)
-    params = parse_qs(parsed_url.query)
-    logger.debug(f"Parsed URL parameters: {params}")  # Debugging information
-    return params.get('username', ["Guest"])[0]
-
-def get_username_from_url(url):
-    # Extract username from the actual URL
-    username = extract_username_from_url(url)
-    logger.info(f"Extracted username: {username}")
-    return username
-
-# Replace this URL with your actual URL where the Gradio app is hosted
-url = 'https://measmonysuon-imagen.hf.space/?username=Guest_01'  # This should be dynamically obtained if needed
-username = get_username_from_url(url)
+# Example URL (this should be dynamically obtained in your use case)
+url = 'https://measmonysuon-imagen.hf.space/?username=Guest_001'
+username = extract_username_from_url(url)
+logger.info(f"Extracted username: {username}")
 launch_gradio(username)