Update app.py

app.py

@@ -1,157 +1,80 @@
 import os
-import logging
-import tempfile
-import httpx
+import telebot
+import requests
 from fastapi import FastAPI, Request
 
-# -------------------------
-# CONFIG
-# -------------------------
-BOT_TOKEN = "8749818258:AAFls-Y46BXW-b94E8-HQTDCMosYfeEmB8E"
-API_BASE = f"https://api.telegram.org/bot{BOT_TOKEN}"
-FILE_BASE = f"https://api.telegram.org/file/bot{BOT_TOKEN}"
+TOKEN = "8749818258:AAFls-Y46BXW-b94E8-HQTDCMosYfeEmB8E"
+bot = telebot.TeleBot(TOKEN)
 
 app = FastAPI()
-logging.basicConfig(level=logging.INFO)
 
 user_files = {}
 
-
-# -------------------------
-# Telegram API helper
-# -------------------------
-async def tg_api(method, data=None, files=None):
-    url = f"{API_BASE}/{method}"
-    async with httpx.AsyncClient(timeout=60) as client:
-        if files:
-            r = await client.post(url, data=data, files=files)
-        else:
-            r = await client.post(url, json=data)
-    return r.json()
-
-
-# -------------------------
-# Get File Path
-# -------------------------
-async def get_file_path(file_id):
-    res = await tg_api("getFile", {"file_id": file_id})
-    return res["result"]["file_path"]
-
-
-# -------------------------
-# Download File
-# -------------------------
-async def download_file(file_path):
-    file_url = f"{FILE_BASE}/{file_path}"
-    async with httpx.AsyncClient(timeout=120) as client:
-        r = await client.get(file_url)
-    tmp = tempfile.NamedTemporaryFile(delete=False)
-    tmp.write(r.content)
-    tmp.close()
-    return tmp.name
-
-
-# -------------------------
-# Webhook
-# -------------------------
+# -----------------------
+# FILE HANDLER
+# -----------------------
+@bot.message_handler(content_types=['document','video','audio','photo','voice','animation'])
+def handle_file(message):
+    user_files[message.from_user.id] = message
+
+    bot.send_message(
+        message.chat.id,
+        "File received.\nReply with:\n"
+        "document / video / audio / photo / voice / animation"
+    )
+
+# -----------------------
+# CONVERSION HANDLER
+# -----------------------
+@bot.message_handler(func=lambda m: True)
+def convert(message):
+    desired = message.text.lower()
+
+    if message.from_user.id not in user_files:
+        bot.send_message(message.chat.id, "Send a file first.")
+        return
+
+    original = user_files[message.from_user.id]
+    file_id = None
+
+    if original.document:
+        file_id = original.document.file_id
+    elif original.video:
+        file_id = original.video.file_id
+    elif original.audio:
+        file_id = original.audio.file_id
+    elif original.photo:
+        file_id = original.photo[-1].file_id
+    elif original.voice:
+        file_id = original.voice.file_id
+    elif original.animation:
+        file_id = original.animation.file_id
+
+    file_info = bot.get_file(file_id)
+    file_url = f"https://api.telegram.org/file/bot{TOKEN}/{file_info.file_path}"
+
+    file_bytes = requests.get(file_url).content
+
+    if desired == "video":
+        bot.send_video(message.chat.id, file_bytes)
+    elif desired == "audio":
+        bot.send_audio(message.chat.id, file_bytes)
+    elif desired == "document":
+        bot.send_document(message.chat.id, file_bytes)
+    elif desired == "photo":
+        bot.send_photo(message.chat.id, file_bytes)
+    elif desired == "voice":
+        bot.send_voice(message.chat.id, file_bytes)
+    elif desired == "animation":
+        bot.send_animation(message.chat.id, file_bytes)
+    else:
+        bot.send_message(message.chat.id, "Unknown format.")
+
+# -----------------------
+# WEBHOOK ROUTE
+# -----------------------
 @app.post("/webhook")
 async def webhook(request: Request):
-    update = await request.json()
-    logging.info(update)
-
-    if "message" not in update:
-        return {"ok": True}
-
-    message = update["message"]
-    chat_id = message["chat"]["id"]
-    user_id = message["from"]["id"]
-
-    # -------------------------
-    # Commands
-    # -------------------------
-    if "text" in message and message["text"].startswith("/"):
-        cmd = message["text"]
-
-        if cmd == "/start":
-            await tg_api("sendMessage", {
-                "chat_id": chat_id,
-                "text": "Send a file, then reply with:\n"
-                        "document / audio / video / photo / voice / animation"
-            })
-            return {"ok": True}
-
-        if cmd == "/reset":
-            user_files.pop(user_id, None)
-            await tg_api("sendMessage", {
-                "chat_id": chat_id,
-                "text": "Session cleared."
-            })
-            return {"ok": True}
-
-    # -------------------------
-    # File Reception
-    # -------------------------
-    media_types = ["document", "audio", "video", "photo", "voice", "animation"]
-
-    for media in media_types:
-        if media in message:
-            if media == "photo":
-                file_id = message["photo"][-1]["file_id"]
-            else:
-                file_id = message[media]["file_id"]
-
-            user_files[user_id] = file_id
-
-            await tg_api("sendMessage", {
-                "chat_id": chat_id,
-                "text": f"Received as {media}.\nNow reply with desired method."
-            })
-
-            return {"ok": True}
-
-    # -------------------------
-    # Conversion
-    # -------------------------
-    if "text" in message:
-        desired = message["text"].lower().strip()
-
-        if user_id not in user_files:
-            await tg_api("sendMessage", {
-                "chat_id": chat_id,
-                "text": "Send a file first."
-            })
-            return {"ok": True}
-
-        file_id = user_files[user_id]
-
-        method_map = {
-            "document": "sendDocument",
-            "audio": "sendAudio",
-            "video": "sendVideo",
-            "photo": "sendPhoto",
-            "voice": "sendVoice",
-            "animation": "sendAnimation"
-        }
-
-        if desired not in method_map:
-            await tg_api("sendMessage", {
-                "chat_id": chat_id,
-                "text": "Unknown format."
-            })
-            return {"ok": True}
-
-        # Step 1: Get file path
-        file_path = await get_file_path(file_id)
-
-        # Step 2: Download file
-        local_file = await download_file(file_path)
-
-        # Step 3: Re-upload as desired type
-        with open(local_file, "rb") as f:
-            await tg_api(method_map[desired], {
-                "chat_id": chat_id
-            }, files={desired: f})
-
-        return {"ok": True}
-
+    update = telebot.types.Update.de_json(await request.json())
+    bot.process_new_updates([update])
     return {"ok": True}