Create app.py

app.py

@@ -0,0 +1,119 @@
+import logging
+import os
+from fastapi import FastAPI, Request
+from contextlib import asynccontextmanager
+from telegram import Update
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackContext
+import httpx
+import asyncio
+from transformers import pipeline
+from langdetect import detect
+from huggingface_hub import login
+
+
+# Global variables
+TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")  # Telegram Token
+HF_HUB_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+
+
+# Verify Hugging Face token
+if not HF_HUB_TOKEN:
+    raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN in environment variables.")
+login(token=HF_HUB_TOKEN)
+
+
+# Load Hebrew and English text generation models
+hebrew_generator = pipeline("text-generation", model="Norod78/hebrew-gpt_neo-small")
+english_generator = pipeline("text-generation", model="distilgpt2")
+
+
+# Function to detect language
+def detect_language(user_input):
+    try:
+        lang = detect(user_input)
+        return "hebrew" if lang == "he" else "english" if lang == "en" else "unsupported"
+    except:
+        return "unsupported"
+
+
+# Function to generate a response
+def generate_response(text):
+    language = detect_language(text)
+    if language == "hebrew":
+        return hebrew_generator(text, max_length=100)[0]["generated_text"]
+    elif language == "english":
+        return english_generator(text, max_length=100)[0]["generated_text"]
+    return "Sorry, I only support Hebrew and English."
+
+
+# Create Telegram Bot
+telegram_app = Application.builder().token(TOKEN).build()
+
+
+# Initialize Telegram
+async def init_telegram():
+    await telegram_app.initialize()
+    await telegram_app.start()
+
+
+# Configure logging
+logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+# Command and Message Handlers
+async def start(update: Update, context: CallbackContext):
+    await update.message.reply_text("Hello! Tell me your decision-making issue in Hebrew or English, and I'll try to help you.")
+
+
+async def handle_message(update: Update, context: CallbackContext):
+    user_text = update.message.text
+    response = generate_response(user_text)
+    await update.message.reply_text(response)
+
+
+telegram_app.add_handler(CommandHandler("start", start))
+telegram_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+
+
+# FastAPI lifespan event
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    print("Starting application...")
+
+    # Check internet connection
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.get("https://api.telegram.org")
+            print(f"Network test successful! Status Code: {response.status_code}")
+    except Exception as e:
+        print(f"Network test failed: {e}")
+
+    # Start Telegram bot
+    asyncio.create_task(init_telegram())  # Run in background
+
+    yield  # Wait until app closes
+    
+    print("Shutting down application...")
+
+
+# Create FastAPI app
+app = FastAPI(lifespan=lifespan)
+
+
+@app.get("/")
+async def root():
+    return {"message": "Decision Helper Bot is running!"}
+
+
+@app.post("/")
+async def receive_update(request: Request):
+    update = Update.de_json(await request.json(), telegram_app.bot)
+    await telegram_app.process_update(update)
+    return {"status": "ok"}
+
+
+# Run the server
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)