Create main.py

main.py

@@ -0,0 +1,155 @@
+import os
+import logging
+import asyncio
+import nest_asyncio
+
+# Import the Hugging Face Inference Client and login function
+from huggingface_hub import InferenceClient, login
+import langid
+
+# Import Telegram bot components
+from telegram import Update, Bot
+from telegram.ext import Application, CommandHandler, MessageHandler, filters, CallbackContext
+
+# -------------------------
+# Configure logging
+# -------------------------
+logging.basicConfig(
+    format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+# -------------------------
+# Environment variables and login to Hugging Face
+# -------------------------
+HF_HUB_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+if not HF_HUB_TOKEN:
+    raise ValueError("Missing Hugging Face API token. Please set HUGGINGFACEHUB_API_TOKEN.")
+
+login(token=HF_HUB_TOKEN)
+client = InferenceClient(api_key=HF_HUB_TOKEN)
+
+# Get Telegram Bot Token
+TOKEN = os.getenv("TELEGRAM_BOT_TOKEN")
+if not TOKEN:
+    raise ValueError("Missing Telegram Bot Token. Please set TELEGRAM_BOT_TOKEN.")
+
+# Get Webhook Domain (e.g., your-space.hf.space)
+WEBHOOK_DOMAIN = os.getenv("WEBHOOK_DOMAIN")
+if not WEBHOOK_DOMAIN:
+    raise ValueError("Missing Webhook Domain. Please set WEBHOOK_DOMAIN.")
+
+# Construct the full webhook URL.
+# We use the Telegram bot token as a unique path.
+WEBHOOK_URL = f"https://{WEBHOOK_DOMAIN}/{TOKEN}"
+
+
+# -------------------------
+# Function to detect language
+# -------------------------
+def detect_language(user_input: str):
+    try:
+        lang, _ = langid.classify(user_input)
+        if lang == "he":
+            return "hebrew"
+        elif lang == "en":
+            return "english"
+        else:
+            return "unsupported"
+    except Exception as e:
+        logger.error(f"Language detection error: {e}")
+        return "unsupported"
+
+
+# -------------------------
+# Function to generate response using Hugging Face Chat models
+# -------------------------
+def generate_response(text: str) -> str:
+    language = detect_language(text)
+    
+    if language == "hebrew":
+        # For Hebrew, prompt the model accordingly.
+        content = "תענה בקצרה אבל תשתף את תהליך קבלת ההחלטות שלך, " + text
+        model = "microsoft/Phi-3.5-mini-instruct"
+    elif language == "english":
+        # For English, use a different prompt.
+        content = "keep it short but tell your decision making process, " + text
+        model = "mistralai/Mistral-Nemo-Instruct-2407"
+    else:
+        return "Sorry, I only support Hebrew and English."
+    
+    messages = [{"role": "user", "content": content}]
+    
+    try:
+        # Call the chat completion API
+        completion = client.chat.completions.create(
+            model=model,
+            messages=messages,
+            max_tokens=2048,
+            temperature=0.5,
+            top_p=0.7
+        )
+        return completion.choices[0].message.content
+    except Exception as e:
+        logger.error(f"Error generating response: {e}")
+        return "Error: Could not generate response."
+
+
+# -------------------------
+# Telegram Bot Handlers
+# -------------------------
+async def start(update: Update, context: CallbackContext):
+    # Respond to the /start command.
+    await update.message.reply_text("Hello! Tell me your decision-making issue, and I'll try to help.")
+    logger.info("Start command received.")
+
+
+async def handle_message(update: Update, context: CallbackContext):
+    user_text = update.message.text
+    logger.info(f"User message: {user_text}")
+    
+    # Generate a response using our Hugging Face chat logic.
+    response_text = generate_response(user_text)
+    logger.info(f"Generated response: {response_text}")
+    
+    await update.message.reply_text(response_text)
+
+
+# -------------------------
+# Main function to run the Telegram Bot using Webhook mode
+# -------------------------
+async def main():
+    # Build the Application with the Telegram Bot Token.
+    application = Application.builder().token(TOKEN).build()
+
+    # Add command and message handlers.
+    application.add_handler(CommandHandler("start", start))
+    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
+
+    logger.info("Starting bot in webhook mode...")
+    print("Starting bot in webhook mode...")
+
+    # Run the bot in webhook mode.
+    # The bot will listen on all interfaces (0.0.0.0) at port 7860.
+    # Telegram will send updates to WEBHOOK_URL.
+    await application.run_webhook(
+        listen="0.0.0.0",         # Listen on all interfaces.
+        port=7860,                # Port to listen on.
+        url_path=TOKEN,           # Use the bot token as the URL path.
+        webhook_url=WEBHOOK_URL   # Full webhook URL.
+    )
+
+
+# -------------------------
+# Run the main function without closing a running event loop
+# -------------------------
+if __name__ == "__main__":
+    # Apply nest_asyncio to allow nested event loops (needed in some HF Space environments).
+    nest_asyncio.apply()
+    loop = asyncio.get_event_loop()
+    
+    try:
+        loop.run_until_complete(main())
+    except Exception as e:
+        logger.error(f"Error in main loop: {e}")
+        print(f"Error in main loop: {e}")