Spaces:
Runtime error
Runtime error
File size: 7,145 Bytes
1d33068 24ea1ea ca3defd a2d5af8 8e64f0d 1d33068 ac77ded dcc49a2 928e72d dcc49a2 8e64f0d 1d33068 8e64f0d 1d33068 8e64f0d 1d33068 a2d5af8 f35c9d9 24ea1ea f35c9d9 24ea1ea e48bfe4 24ea1ea f35c9d9 24ea1ea ac77ded f35c9d9 24ea1ea f35c9d9 24ea1ea af27696 f35c9d9 24ea1ea ca3defd 24ea1ea ca3defd 24ea1ea f35c9d9 ca3defd f35c9d9 ca3defd f35c9d9 a2d5af8 f35c9d9 dea0635 a2d5af8 1d33068 ca3defd 928e72d 9a29b5b dea0635 ca3defd 64536ea dea0635 8e64f0d 64536ea 0277ed0 1563aef 928e72d 1563aef 0277ed0 1563aef f8957c5 1563aef 928e72d 1563aef d0ce7b4 8e64f0d f35c9d9 8e64f0d 1d33068 928e72d 1d33068 8e64f0d 1d33068 ca3defd f5bcaca 1d33068 24ea1ea ca3defd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 |
import gradio as gr
from gradio_client import Client
import os
import logging
import sqlite3
import requests
from urllib.parse import urlparse, parse_qs
# Initialize the client for image generation
client_image = Client("mukaist/DALLE-4K")
# Retrieve secret token from environment variables
webhook_server = os.getenv('webhook_server')
db_path = os.getenv('db_path') # Ensure you have this set up
# Define resolutions
resolutions = {
"896x1152": (896, 1152),
"1024x1024": (1024, 1024),
"1216x832": (1216, 832)
}
# Define the default style
DEFAULT_STYLE = "3840 x 2160"
# Set up logging
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 update_user_points(user_chat_id, points):
"""Updates the user's points in the database."""
with sqlite3.connect(db_path) as conn:
cursor = conn.cursor()
cursor.execute('''
INSERT INTO users (user_chat_id, points) VALUES (?, ?)
ON CONFLICT(user_chat_id) DO UPDATE SET points = points + ?
''', (user_chat_id, points, points))
conn.commit()
logger.info(f"User with ID {user_chat_id} had their points updated by {points}.")
def notify_webhook(user_chat_id):
"""Notify the webhook server about the points update."""
webhook_url = f"{webhook_server}/update_points"
payload = {'user_chat_id': user_chat_id, 'points': 10} # Adjust payload as necessary
try:
response = requests.post(webhook_url, json=payload)
response.raise_for_status()
logger.info(f"Webhook notification sent successfully. Response: {response.status_code}")
except requests.RequestException as e:
logger.error(f"Error sending webhook notification: {e}")
def get_user_points(user_chat_id):
"""Retrieve user points from the Flask server."""
webhook_url = f"{webhook_server}/get_points"
params = {'user_chat_id': user_chat_id}
try:
response = requests.get(webhook_url, params=params)
response.raise_for_status()
return response.json().get('points', "Error retrieving points")
except requests.RequestException as e:
logger.error(f"Error fetching user points: {e}")
return "Failed to retrieve user points"
def extract_user_chat_id_from_url(url):
parsed_url = urlparse(url)
params = parse_qs(parsed_url.query)
return params.get('user_chat_id', ["Guest"])[0]
def gradio_interface(prompt, resolution_key, user_chat_id):
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):
update_user_points(user_chat_id, 10) # Award 10 points for each image generated
notify_webhook(user_chat_id) # Notify the webhook server
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."
def create_gradio_interface():
def update_points_and_check(user_chat_id):
points = get_user_points(user_chat_id)
if points == "Failed to retrieve user points":
return points, gr.Button("Generate", visible=False)
else:
points = int(points)
return points, gr.Button("Generate", visible=points >= 5)
with gr.Blocks() as interface:
# Personalized HTML content
gr.HTML("""
<h3>Welcome! Please enter your user chat ID to continue.</h3>
""")
# Create input components
user_chat_id_input = gr.Textbox(label="User Chat ID", placeholder="Enter your user chat ID here...")
points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
# Create the button to get user points
get_points_button = gr.Button("Get Points")
# Create a component for the button state
generate_button = gr.Button("Generate", visible=False)
# Arrange user_chat_id_input, points_output, and get_points_button in one row
with gr.Row():
user_chat_id_input
points_output
get_points_button
# Arrange generate_button in a separate row
with gr.Row():
generate_button
# Set up interactions
get_points_button.click(
fn=update_points_and_check,
inputs=[user_chat_id_input],
outputs=[points_output, generate_button]
)
generate_button.click(
fn=lambda prompt, resolution_key, user_chat_id: gradio_interface(prompt, resolution_key, user_chat_id),
inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
outputs=[result_output, message_output]
)
# Add other components for prompt input, resolution selection, etc.
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
result_output = gr.Image(label="Generated Image", type="pil")
message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
# Update the generate button's visibility based on the points check
get_points_button.click(
fn=update_points_and_check,
inputs=[user_chat_id_input],
outputs=[points_output, generate_button]
)
gr.HTML("""
<style>
footer.svelte-1rjryqp {
display: none !important;
}
</style>
""")
return interface
def launch_gradio():
interface = create_gradio_interface()
interface.launch()
if __name__ == '__main__':
# Launch Gradio
launch_gradio() |