Spaces:
Runtime error
Runtime error
measmonysuon
commited on
Commit
•
be586af
1
Parent(s):
b3fd847
Update app.py
Browse files
app.py
CHANGED
@@ -121,7 +121,7 @@ def create_gradio_interface():
|
|
121 |
# Create other components
|
122 |
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
|
123 |
resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
|
124 |
-
generate_button = gr.Button("Generate")
|
125 |
|
126 |
result_output = gr.Image(label="Generated Image", type="pil")
|
127 |
message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
|
@@ -137,20 +137,27 @@ def create_gradio_interface():
|
|
137 |
resolution_dropdown
|
138 |
generate_button
|
139 |
|
140 |
-
def
|
141 |
points = get_user_points(user_chat_id)
|
142 |
if points >= 5:
|
143 |
return points, gr.Button.update(visible=True)
|
144 |
else:
|
145 |
return points, gr.Button.update(visible=False)
|
146 |
-
|
147 |
-
def handle_get_points(user_chat_id):
|
148 |
-
points, button_state = update_points_and_generate_button(user_chat_id)
|
149 |
-
return points, button_state
|
150 |
|
151 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
152 |
generate_button.click(
|
153 |
-
fn=
|
154 |
inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
|
155 |
outputs=[result_output, message_output]
|
156 |
)
|
|
|
121 |
# Create other components
|
122 |
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
|
123 |
resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
|
124 |
+
generate_button = gr.Button("Generate", visible=False) # Start with the button hidden
|
125 |
|
126 |
result_output = gr.Image(label="Generated Image", type="pil")
|
127 |
message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
|
|
|
137 |
resolution_dropdown
|
138 |
generate_button
|
139 |
|
140 |
+
def handle_get_points(user_chat_id):
|
141 |
points = get_user_points(user_chat_id)
|
142 |
if points >= 5:
|
143 |
return points, gr.Button.update(visible=True)
|
144 |
else:
|
145 |
return points, gr.Button.update(visible=False)
|
|
|
|
|
|
|
|
|
146 |
|
147 |
+
def handle_generate_image(prompt, resolution_key, user_chat_id):
|
148 |
+
points = get_user_points(user_chat_id)
|
149 |
+
if points >= 5:
|
150 |
+
result = gradio_interface(prompt, resolution_key, user_chat_id)
|
151 |
+
if result[0]: # If image generation is successful
|
152 |
+
return result[0], "The image was generated successfully."
|
153 |
+
else:
|
154 |
+
return None, "There was an error processing your photo. Please try again later."
|
155 |
+
else:
|
156 |
+
return None, "Insufficient points. Please get more points before generating an image."
|
157 |
+
|
158 |
+
# Set up interactions
|
159 |
generate_button.click(
|
160 |
+
fn=handle_generate_image,
|
161 |
inputs=[prompt_input, resolution_dropdown, user_chat_id_input],
|
162 |
outputs=[result_output, message_output]
|
163 |
)
|