Update app.py

app.py

@@ -1,8 +1,10 @@
+from flask import Flask, request, redirect
 import gradio as gr
 from gradio_client import Client
 import os
 import logging
-from urllib.parse import parse_qs, urlparse
+
+app = Flask(__name__)
 
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
@@ -21,14 +23,42 @@ DEFAULT_STYLE = "3840 x 2160"
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# 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]
+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."
 
-# Define the Gradio app
 def create_gradio_interface(username):
     with gr.Blocks() as interface:
         # Personalized HTML content
@@ -52,19 +82,6 @@ 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),
@@ -82,13 +99,12 @@ def create_gradio_interface(username):
     
     return interface
 
-# Launch Gradio app
-def launch_gradio(username):
+@app.route('/gradio')
+def serve_gradio():
+    username = request.args.get('username', 'Guest')
+    logger.info(f"Extracted username: {username}")
     interface = create_gradio_interface(username)
-    interface.launch()
+    return interface.launch(inline=True)
 
-# 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)
+if __name__ == '__main__':
+    app.run(debug=True)