Spaces:
Runtime error
Runtime error
measmonysuon
commited on
Commit
•
928e72d
1
Parent(s):
1563aef
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ client_image = Client("mukaist/DALLE-4K")
|
|
11 |
|
12 |
# Retrieve secret token from environment variables
|
13 |
webhook_server = os.getenv('webhook_server')
|
|
|
14 |
|
15 |
# Define resolutions
|
16 |
resolutions = {
|
@@ -104,6 +105,14 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
|
|
104 |
return None, "There was an error processing your photo. Please try again later."
|
105 |
|
106 |
def create_gradio_interface():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
107 |
with gr.Blocks() as interface:
|
108 |
# Personalized HTML content
|
109 |
gr.HTML("""
|
@@ -118,16 +127,7 @@ def create_gradio_interface():
|
|
118 |
get_points_button = gr.Button("Get Points")
|
119 |
|
120 |
# Create a component for the button state
|
121 |
-
generate_button = gr.Button("Generate")
|
122 |
-
generate_button.style(height=40)
|
123 |
-
|
124 |
-
# Function to check points and enable/disable generate button
|
125 |
-
def check_points_and_update(user_chat_id):
|
126 |
-
points = get_user_points(user_chat_id)
|
127 |
-
if points == "Failed to retrieve user points":
|
128 |
-
return points, gr.Button("Generate", visible=False)
|
129 |
-
else:
|
130 |
-
return points, gr.Button("Generate", visible=points >= 5)
|
131 |
|
132 |
# Arrange user_chat_id_input, points_output, and get_points_button in one row
|
133 |
with gr.Row():
|
@@ -141,7 +141,7 @@ def create_gradio_interface():
|
|
141 |
|
142 |
# Set up interactions
|
143 |
get_points_button.click(
|
144 |
-
fn=
|
145 |
inputs=[user_chat_id_input],
|
146 |
outputs=[points_output, generate_button]
|
147 |
)
|
@@ -152,6 +152,19 @@ def create_gradio_interface():
|
|
152 |
outputs=[result_output, message_output]
|
153 |
)
|
154 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
155 |
gr.HTML("""
|
156 |
<style>
|
157 |
footer.svelte-1rjryqp {
|
|
|
11 |
|
12 |
# Retrieve secret token from environment variables
|
13 |
webhook_server = os.getenv('webhook_server')
|
14 |
+
db_path = os.getenv('db_path') # Ensure you have this set up
|
15 |
|
16 |
# Define resolutions
|
17 |
resolutions = {
|
|
|
105 |
return None, "There was an error processing your photo. Please try again later."
|
106 |
|
107 |
def create_gradio_interface():
|
108 |
+
def update_points_and_check(user_chat_id):
|
109 |
+
points = get_user_points(user_chat_id)
|
110 |
+
if points == "Failed to retrieve user points":
|
111 |
+
return points, gr.Button("Generate", visible=False)
|
112 |
+
else:
|
113 |
+
points = int(points)
|
114 |
+
return points, gr.Button("Generate", visible=points >= 5)
|
115 |
+
|
116 |
with gr.Blocks() as interface:
|
117 |
# Personalized HTML content
|
118 |
gr.HTML("""
|
|
|
127 |
get_points_button = gr.Button("Get Points")
|
128 |
|
129 |
# Create a component for the button state
|
130 |
+
generate_button = gr.Button("Generate", visible=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
131 |
|
132 |
# Arrange user_chat_id_input, points_output, and get_points_button in one row
|
133 |
with gr.Row():
|
|
|
141 |
|
142 |
# Set up interactions
|
143 |
get_points_button.click(
|
144 |
+
fn=update_points_and_check,
|
145 |
inputs=[user_chat_id_input],
|
146 |
outputs=[points_output, generate_button]
|
147 |
)
|
|
|
152 |
outputs=[result_output, message_output]
|
153 |
)
|
154 |
|
155 |
+
# Add other components for prompt input, resolution selection, etc.
|
156 |
+
prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
|
157 |
+
resolution_dropdown = gr.Dropdown(choices=list(resolutions.keys()), label="Resolution", value="1024x1024")
|
158 |
+
result_output = gr.Image(label="Generated Image", type="pil")
|
159 |
+
message_output = gr.Textbox(label="Result", placeholder="Results will be shown here", interactive=False)
|
160 |
+
|
161 |
+
# Update the generate button's visibility based on the points check
|
162 |
+
get_points_button.click(
|
163 |
+
fn=update_points_and_check,
|
164 |
+
inputs=[user_chat_id_input],
|
165 |
+
outputs=[points_output, generate_button]
|
166 |
+
)
|
167 |
+
|
168 |
gr.HTML("""
|
169 |
<style>
|
170 |
footer.svelte-1rjryqp {
|