Update app.py

app.py

@@ -8,7 +8,6 @@ from urllib.parse import urlparse, parse_qs
 # Initialize the client for image generation
 client_image = Client("mukaist/DALLE-4K")
 
-
 # Define resolutions and default style
 resolutions = {
     "896x1152": (896, 1152),
@@ -45,14 +44,17 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
         logger.error(f"Error generating image: {e}")
         return None
 
+def request_otp(user_chat_id):
+    response = requests.post(f"{webhook_server}/request_otp", json={"user_chat_id": user_chat_id})
+    return response.json()
+
 def verify_otp(user_chat_id, otp):
     response = requests.post(f"{webhook_server}/verify_otp", json={"user_chat_id": user_chat_id, "otp": otp})
     return response.json().get("valid", False)
 
 def retrieve_user_points(user_chat_id, otp):
     # Verify OTP via Flask server
-    otp_response = requests.post(f"{webhook_server}/verify_otp", json={"user_chat_id": user_chat_id, "otp": otp})
-    if otp_response.status_code != 200 or not otp_response.json().get('valid'):
+    if not verify_otp(user_chat_id, otp):
         return "Invalid or expired OTP. Please request a new OTP."
 
     # Retrieve user points via Flask server
@@ -132,6 +134,12 @@ def handle_generate_image(prompt, resolution_key, user_chat_id):
         return None, "There was an error processing your photo. Please try again later."
     return None, "Insufficient points. Please get more points before generating an image."
 
+def request_otp_and_update_ui(user_chat_id):
+    otp_response = request_otp(user_chat_id)
+    if otp_response.get("success"):
+        return "OTP has been sent to your registered contact."
+    return "Error requesting OTP. Please try again later."
+
 def create_gradio_interface():
     with gr.Blocks() as interface:
         gr.HTML("""
@@ -141,8 +149,12 @@ def create_gradio_interface():
 
         user_chat_id_input = gr.Textbox(label="Your UID", placeholder="Your UID")
         user_otp_input = gr.Textbox(label="OTP", type="password")
-        points_output = gr.Textbox(label="Your Balances", placeholder="Your Points", interactive=False)
         
+        with gr.Row():
+            request_otp_button = gr.Button("Request OTP")
+            user_otp_input
+        
+        points_output = gr.Textbox(label="Your Balances", placeholder="Your Points", interactive=False)
         
         get_points_button = gr.Button("Show Your Points")
         prompt_input = gr.Textbox(label="Prompt", placeholder="Enter your prompt here...")
@@ -155,6 +167,7 @@ def create_gradio_interface():
         with gr.Row():
             user_chat_id_input
             points_output
+            request_otp_button
             get_points_button
             
         with gr.Row():
@@ -174,6 +187,12 @@ def create_gradio_interface():
             outputs=[points_output]
         )
         
+        request_otp_button.click(
+            fn=request_otp_and_update_ui,
+            inputs=[user_chat_id_input],
+            outputs=[message_output]
+        )
+        
         gr.HTML("""
         <style>
             footer.svelte-1rjryqp {