Update app.py

app.py

@@ -176,30 +176,19 @@ def get_files_api():
 
 @app.route('/api/add-file', methods=['POST'])
 def add_file_api():
-    file_id = request.form.get('file_id')
-    file_ref = request.form.get('file_ref')
-    file_name = request.form.get('file_name')
-    file_size = request.form.get('file_size')
-    file_type = request.form.get('file_type')
-    mime_type = request.form.get('mime_type')
-    caption = request.form.get('caption')
-    if not file_id or not file_name or not file_size:
-        return jsonify({"error": "File ID, file name, and file size are required"}), 400
-    file = Media(
-        file_id=file_id,
-        file_ref=file_ref,
-        file_name=file_name,
-        file_size=int(file_size),
-        file_type=file_type,
-        mime_type=mime_type,
-        caption=caption
-    )
-    try:
+    if 'fileUpload' not in request.files:
+        return jsonify({"error": "No file part"}), 400
+    file = request.files['fileUpload']
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+    if file:
+        file_path = os.path.join("/app/uploads", file.filename)
+        file.save(file_path)
         loop = asyncio.get_event_loop()
-        loop.run_until_complete(file.commit())
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-    return jsonify({"message": "File added successfully"})
+        file_id, file_ref = loop.run_until_complete(save_file(file_path))
+        os.remove(file_path)
+        return jsonify({"file_id": file_id, "file_ref": file_ref, "message": "File uploaded successfully"})
+    return jsonify({"error": "Failed to upload file"}), 500
 
 @app.route('/api/delete-file', methods=['POST'])
 def delete_file_api():
@@ -336,10 +325,22 @@ def broadcast_api():
     deleted = 0
     failed = 0
     success = 0
-    loop.run_until_complete(asyncio.gather(*[
-        asyncio.create_task(handle_user(user, message_text, success, blocked, deleted, failed, done))
-        for user in users
-    ]))
+
+    async def handle_user(user):
+        nonlocal success, blocked, deleted, failed, done
+        try:
+            await bot.send_message(user['id'], message_text)
+            success += 1
+        except UserIsBlocked:
+            blocked += 1
+        except InputUserDeactivated:
+            deleted += 1
+        except Exception as e:
+            failed += 1
+        done += 1
+
+    loop.run_until_complete(asyncio.gather(*(handle_user(user) for user in users)))
+
     return jsonify({
         "total_users": total_users,
         "completed": done,
@@ -349,18 +350,6 @@ def broadcast_api():
         "failed": failed
     })
 
-async def handle_user(user, message_text, success, blocked, deleted, failed, done):
-    try:
-        await bot.send_message(user['id'], message_text)
-        success += 1
-    except UserIsBlocked:
-        blocked += 1
-    except InputUserDeactivated:
-        deleted += 1
-    except Exception as e:
-        failed += 1
-    done += 1
-
 @app.route('/api/ban-users')
 def ban_users_api():
     loop = asyncio.get_event_loop()
@@ -406,11 +395,12 @@ def chat_info_api(chat_id):
     try:
         loop = asyncio.get_event_loop()
         chat = loop.run_until_complete(bot.get_chat(int(chat_id)))
+        members_count = loop.run_until_complete(bot.get_chat_members_count(int(chat_id)))
         return jsonify({
             "title": chat.title,
             "username": chat.username,
             "id": chat.id,
-            "members_count": loop.run_until_complete(bot.get_chat_members_count(int(chat_id)))
+            "members_count": members_count
         })
     except Exception as e:
         return jsonify({"error": str(e)}), 500
@@ -507,4 +497,5 @@ def connections_api():
     return jsonify({"connections": connections})
 
 if __name__ == '__main__':
+    bot.run()
     app.run(host='0.0.0.0', port=7860, debug=False)