Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 from huggingface_hub import login, whoami, HfFolder, HfApi, SpaceHardware
 import time
 import torch
+import json 
 
 webhook_server = os.getenv('webhook_server')
 API_TOKEN = os.getenv('ZeroGPU')
@@ -69,7 +70,18 @@ def generate_image(prompt, resolution_key, style=DEFAULT_STYLE):
     except Exception as e:
         logger.error(f"Error generating image: {e}")
         return None
-        
+
+def format_gpu_info(gpu_info):
+    # Extract and format GPU information
+    is_cuda_available = gpu_info.get('is_cuda_available', False)
+    device_name = gpu_info.get('device_name', 'Unknown Device')
+    
+    # Create a friendly message
+    cuda_status = "Available" if is_cuda_available else "Not Available"
+    formatted_info = f"Device: {device_name}, CUDA: {cuda_status}"
+    
+    return formatted_info
+    
 def gradio_interface(prompt, resolution_key, user_chat_id):
     if not user_chat_id.strip():
         return None, "User Chat ID cannot be blank. Please provide a valid User Chat ID or register with @supportBot to get UID."
@@ -94,11 +106,14 @@ def gradio_interface(prompt, resolution_key, user_chat_id):
                 full_file_url = base_url + file_path
                 
                 if os.path.exists(file_path):
+                    # Format GPU info
+                    formatted_gpu_info = format_gpu_info(gpu_info)
+                    
                     # Update user points and notify webhook
                     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
+                    notify_webhook_with_file(user_chat_id, full_file_url, formatted_gpu_info)  # Notify webhook with full file URL
                     
-                    return full_file_url, f"Check your telegram, The image was generated successfully. GPU Info: {json.dumps(gpu_info)}"
+                    return full_file_url, f"Check your telegram, The image was generated successfully. GPU Info: {formatted_gpu_info}"
                 
                 return None, "The image file is not available. Please try again later."
             
@@ -130,9 +145,9 @@ def handle_generate_image(prompt, resolution_key, user_chat_id):
     if points >= 5:
         result = gradio_interface(prompt, resolution_key, user_chat_id)
         if result[0]:
-            # Include the GPU info in the success message
-            full_file_url, gpu_info = result
-            return full_file_url, f"Check your telegram, The image was generated successfully. GPU Info: {json.dumps(gpu_info)}"
+            # Include the formatted GPU info in the success message
+            full_file_url, gpu_info_message = result
+            return full_file_url, f"Image generated and sent successfully to telegram. By {gpu_info_message}"
         
         return None, "There was an error processing your photo. Please try again later."