Update app.py

app.py

@@ -82,6 +82,17 @@ def format_gpu_info(gpu_info):
     
     return formatted_info
     
+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."
@@ -245,21 +256,6 @@ def notify_webhook_with_file(user_chat_id, full_file_url, gpu_info):
         logger.info(f"Webhook notification sent successfully for user {user_chat_id}. Response: {response.status_code}")
     except requests.RequestException as e:
         logger.error(f"Error sending webhook notification with file URL: {e}")
-'''        
-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."
-
-    result = generate_image(prompt, resolution_key)
-    if result and result[0]:
-        file_path = result[0][0].get('image')
-        if file_path and os.path.exists(file_path):
-            update_user_points(user_chat_id, -5)
-            notify_webhook_with_file(user_chat_id, file_path)  # Notify webhook with file path
-            return file_path, "The image was generated successfully."
-        return None, "The image file is not available. Please try again later."
-    return None, "There was an error processing your photo. Please try again later."
-'''
 
 def request_otp_and_update_ui(user_chat_id):
     # Validate user_chat_id