Update app.py

app.py

@@ -79,16 +79,13 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
     
     # Add a short delay to allow the hardware switch to complete
     time.sleep(5)  # Adjust this value if needed
-    
-    # Get GPU info
-    gpu_info = get_gpu_info()
 
     try:
         # Generate the image
-        result = generate_image(prompt, resolution_key)
+        image_info, seed, gpu_info = generate_image(prompt, resolution_key)
         
-        if result and result[0]:
-            file_path = result[0][0].get('image')
+        if image_info and image_info[0]:
+            file_path = image_info[0].get('image')
             if file_path:
                 # Combine base URL with file path to create full URL
                 base_url = "https://measmonysuon-imagen.hf.space/file="
@@ -99,7 +96,7 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
                     update_user_points(user_chat_id, -5)
                     notify_webhook_with_file(user_chat_id, full_file_url, gpu_info)  # Notify webhook with full file URL
                     
-                    return full_file_url, f"Check your telegram, The image was generated successfully. {gpu_info}"
+                    return full_file_url, f"Check your telegram, The image was generated successfully. GPU Info: {json.dumps(gpu_info)}"
                 
                 return None, "The image file is not available. Please try again later."