Update app.py

app.py

@@ -131,29 +131,12 @@ def handle_generate_image(prompt, resolution_key, user_chat_id):
         result = gradio_interface(prompt, resolution_key, user_chat_id)
         if result[0]:
             # Include the GPU info in the success message
-            gpu_info = result[1]  # Assuming gradio_interface returns GPU info in the second part of the tuple
-            return result[0], f"Check your telegram, The image was generated successfully. {gpu_info}"
+            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)}"
         
         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 get_gpu_info():
-    if torch.cuda.is_available():
-        device_name = torch.cuda.get_device_name(0)
-        device_properties = torch.cuda.get_device_properties(0)
-        total_memory = f"{device_properties.total_memory // (1024 ** 2)} MB"
-        return {
-            'is_cuda_available': True,
-            'device_name': device_name,
-            'total_memory': total_memory
-        }
-    else:
-        return {
-            'is_cuda_available': False,
-            'device_name': 'N/A',
-            'total_memory': 'N/A'
-        }
         
 def request_otp(user_chat_id):
     try: