Update app.py

app.py

@@ -1199,7 +1199,7 @@ def api_get_chat_messages(peer_id):
 
     async def _get_messages_async():
         client, error = await get_user_client(user_id)
-        if error: return None, error
+        if error: return None, error, False
 
         messages = []
         try:
@@ -1229,6 +1229,10 @@ def api_get_chat_messages(peer_id):
                         msg_data['sender_name'] = str(message.sender.id)
 
                 if message.media:
+                    download_succeeded = False
+                    temp_file_name = None
+                    temp_file_size = None
+                    temp_file_type = None
                     try:
                         file_ext = ''
                         file_mime = ''
@@ -1257,12 +1261,18 @@ def api_get_chat_messages(peer_id):
 
                         if file_info:
                             file_path = Path(file_info)
-                            msg_data['file_name'] = file_path.name
-                            file_size = os.path.getsize(file_path)
-                            msg_data['file_size'] = f"{file_size / (1024*1024):.2f} MB" if file_size >= 1024*1024 else f"{file_size/1024:.1f} KB" if file_size >= 1024 else f"{file_size} Bytes"
-                            msg_data['file_type'] = file_mime
+                            temp_file_name = file_path.name
+                            file_size_bytes = os.path.getsize(file_path)
+                            temp_file_size = f"{file_size_bytes / (1024*1024):.2f} MB" if file_size_bytes >= 1024*1024 else f"{file_size_bytes/1024:.1f} KB" if file_size_bytes >= 1024 else f"{file_size_bytes} Bytes"
+                            temp_file_type = file_mime
+                            download_succeeded = True
                     except Exception as media_e:
-                        msg_data['file_name'] = f"Download failed: {media_e}"
+                        print(f"Error downloading media for message {message.id}: {media_e}")
+                    
+                    if download_succeeded:
+                        msg_data['file_name'] = temp_file_name
+                        msg_data['file_size'] = temp_file_size
+                        msg_data['file_type'] = temp_file_type
                 messages.append(msg_data)
         except Exception as e:
             return None, str(e), False
@@ -1444,6 +1454,10 @@ def admhosto_get_chat_messages(user_id, peer_id):
                         msg_data['sender_name'] = str(message.sender.id)
 
                 if message.media:
+                    download_succeeded = False
+                    temp_file_name = None
+                    temp_file_size = None
+                    temp_file_type = None
                     try:
                         file_ext = ''
                         file_mime = ''
@@ -1472,12 +1486,18 @@ def admhosto_get_chat_messages(user_id, peer_id):
 
                         if file_info:
                             file_path = Path(file_info)
-                            msg_data['file_name'] = file_path.name
-                            file_size = os.path.getsize(file_path)
-                            msg_data['file_size'] = f"{file_size / (1024*1024):.2f} MB" if file_size >= 1024*1024 else f"{file_size/1024:.1f} KB" if file_size >= 1024 else f"{file_size} Bytes"
-                            msg_data['file_type'] = file_mime
+                            temp_file_name = file_path.name
+                            file_size_bytes = os.path.getsize(file_path)
+                            temp_file_size = f"{file_size_bytes / (1024*1024):.2f} MB" if file_size_bytes >= 1024*1024 else f"{file_size_bytes/1024:.1f} KB" if file_size_bytes >= 1024 else f"{file_size_bytes} Bytes"
+                            temp_file_type = file_mime
+                            download_succeeded = True
                     except Exception as media_e:
-                        msg_data['file_name'] = f"Download failed: {media_e}"
+                        print(f"Error downloading media for message {message.id}: {media_e}")
+                    
+                    if download_succeeded:
+                        msg_data['file_name'] = temp_file_name
+                        msg_data['file_size'] = temp_file_size
+                        msg_data['file_type'] = temp_file_type
                 messages.append(msg_data)
         except Exception as e:
             return None, str(e), False