File size: 6,688 Bytes
1d33068
 
 
 
24ea1ea
 
ca3defd
a2d5af8
8e64f0d
1d33068
8e64f0d
 
1d33068
 
 
 
 
8e64f0d
 
1d33068
8e64f0d
 
1d33068
 
 
f35c9d9
 
 
a2d5af8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f35c9d9
24ea1ea
 
 
 
f35c9d9
 
 
24ea1ea
e48bfe4
24ea1ea
f35c9d9
24ea1ea
ff897dd
f35c9d9
24ea1ea
 
 
 
 
 
 
f35c9d9
24ea1ea
af27696
f35c9d9
24ea1ea
 
 
ca3defd
24ea1ea
 
ca3defd
24ea1ea
f35c9d9
ca3defd
 
f35c9d9
ca3defd
f35c9d9
a2d5af8
 
 
 
 
f35c9d9
 
dea0635
a2d5af8
 
 
 
1d33068
ca3defd
9a29b5b
dea0635
ca3defd
dea0635
 
ca3defd
dea0635
ca3defd
e48bfe4
 
 
ca3defd
dea0635
ca3defd
dea0635
 
8e64f0d
 
 
 
f8957c5
8e64f0d
 
 
 
24ea1ea
 
 
 
f35c9d9
e48bfe4
d0ce7b4
8e64f0d
 
f35c9d9
 
8e64f0d
 
24ea1ea
 
f35c9d9
 
24ea1ea
 
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
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")

# 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__)

# Database path
db_path = 'C:/Users/Administrator/Desktop/TGBoot/photo2prompt_db/users.db'

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 = "http://65.108.76.174:5000/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():
    with gr.Blocks() as interface:
        # Personalized HTML content
        gr.HTML("""
        <script src="https://telegram.org/js/telegram-web-app.js"></script>
        <script>
            document.addEventListener('DOMContentLoaded', function() {
                window.Telegram.WebApp.init();
                const urlParams = new URLSearchParams(window.location.search);
                const userChatId = urlParams.get('user_chat_id') || "Guest";
                window.Telegram.WebApp.setTitle(`Welcome!`);
                document.getElementById('user-chat-id-input').value = userChatId;  // Set hidden textbox with userChatId
            });
        </script>
        <h3>Welcome! Generate images based on prompts.</h3>
        """)

        # Create input components
        prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
        resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
        generate_button = gr.Button("Generate")
        
        # Create output components
        result_output = gr.Image(label="Generated Image", type="pil")
        message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
        
        # Button to get user points
        get_points_button = gr.Button("Get Points")
        points_output = gr.Textbox(label="User Points", placeholder="User points will be displayed here", interactive=False)
        
        # Hidden textbox to hold the user_chat_id
        user_chat_id_input = gr.Textbox(visible=False, value="Guest", elem_id="user-chat-id-input")

        # Set up interaction
        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]
        )

        get_points_button.click(
            fn=lambda user_chat_id: get_user_points(user_chat_id),
            inputs=[user_chat_id_input],
            outputs=points_output
        )
        
        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()