Update app.py

app.py

@@ -1,82 +1,43 @@
-import telebot
-from fastapi import FastAPI, Request, HTTPException
-import uvicorn
-from pydantic import BaseModel
+import os
+import streamlit as st
+from telegram import Update
+from telegram.ext import Application, CommandHandler, ContextTypes
+import logging
+import asyncio
+import threading
 
-# Your bot token is needed internally to send messages via Telegram's API.
-TOKEN = "6655373829:AAGduLdLyNx7zUtxH73Sp3Z1vHKS35tV9WU"
-# Your externally configured webhook URL (set up previously via BotFather or API call)
-WEBHOOK_URL = "https://astraos-testing.hf.space/webhook"
-
-# Initialize the TeleBot instance
-bot = telebot.TeleBot(TOKEN, parse_mode="HTML")
-
-# Create a FastAPI app instance
 app = FastAPI()
-
-# --- Models for request validation ---
-class MessageRequest(BaseModel):
-    chat_id: int
-    text: str
-
-# --- Endpoints ---
-
-@app.post("/webhook")
-async def telegram_webhook(request: Request):
-    """
-    This endpoint receives incoming update payloads from Telegram.
-    Telegram sends updates as JSON via a POST request to this URL.
-    """
-    try:
-        json_data = await request.json()
-    except Exception as e:
-        raise HTTPException(status_code=400, detail="Invalid JSON payload")
-
-    update = telebot.types.Update.de_json(json_data)
-    # Process the update (this will call any message handlers you've set up)
-    bot.process_new_updates([update])
-    # Acknowledge receipt of the update
-    return {"status": "ok"}
-
-@app.get("/")
-def home():
-    """
-    A simple endpoint to verify that your bot’s webhook server is running.
-    """
-    return {"message": "Bot is running via webhook"}
-
-@app.post("/send_message")
-async def send_message(msg: MessageRequest):
-    """
-    This endpoint allows you to send a message from your bot to a user.
-    You provide a JSON payload with 'chat_id' and 'text'.
-    
-    For example, you can POST:
-    
-        {
-          "chat_id": 123456789,
-          "text": "Hello from webhook-only mode!"
-        }
-    
-    The message is sent by calling the Telegram Bot API internally.
-    """
-    try:
-        sent_message = bot.send_message(msg.chat_id, msg.text)
-        return {"status": "sent", "message_id": sent_message.message_id}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
-
-# --- Example Message Handler ---
-@bot.message_handler(commands=["start"])
-def start_command(message):
-    """
-    When a user sends /start to your bot via Telegram (received via webhook),
-    this handler sends a welcome message.
-    """
-    bot.reply_to(message, "Hello! I'm running via FastAPI webhook and ready to send messages!")
-
-# --- Running the App ---
-if __name__ == "__main__":
-    # We run the FastAPI app using uvicorn. Since your webhook is already set up on Telegram,
-    # you only need to start the server to receive updates.
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+# Enable logging
+logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Load bot token
+BOT_TOKEN = "6655373829:AAGduLdLyNx7zUtxH73Sp3Z1vHKS35tV9WU"
+if not BOT_TOKEN:
+    raise ValueError("Bot token is not set in environment variables")
+
+# Create the bot application
+application = Application.builder().token(BOT_TOKEN).build()
+
+# Define the /start command handler
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
+    await update.message.reply_text("Hello! This bot is running on Hugging Face Spaces 🚀")
+
+# Add command handlers
+application.add_handler(CommandHandler("start", start))
+
+# Function to run polling in a thread-safe way
+def run_polling():
+    loop = asyncio.new_event_loop()  # Create a new event loop
+    asyncio.set_event_loop(loop)    # Set it for the current thread
+    loop.run_until_complete(application.run_polling(close_loop=True))  # Run polling in the loop
+
+# Streamlit UI
+st.title("Telegram Bot on Streamlit")
+st.write("This bot is running using Streamlit and Python-Telegram-Bot.")
+
+# Button to start the bot
+if st.button("Start Bot"):
+    thread = threading.Thread(target=run_polling, daemon=True)  # Run polling in a background thread
+    thread.start()
+    st.write("Bot started successfully and is now polling Telegram!")