Update app.py

app.py

@@ -1,73 +1,58 @@
+import os
 import logging
-import threading
-import google.generativeai as genai
-from telegram import Update
-from telegram.ext import Application, CommandHandler, MessageHandler, filters, ContextTypes
 from flask import Flask
+from telegram import Update
+from telegram.ext import ApplicationBuilder, CommandHandler, MessageHandler, ContextTypes, filters
+import google.generativeai as genai
 
 # ==== CONFIG ====
 TELEGRAM_TOKEN = "7745816717:AAGKTpRtuPknjRAIct_2kdoANpJx3ZFztrg"
-GEMINI_API_KEY = "AIzaSyCq23lcvpPfig6ifq1rmt-z11vKpMvDD4I"  # replace with your Gemini key
+GEMINI_API_KEY = "AIzaSyCq23lcvpPfig6ifq1rmt-z11vKpMvDD4I"
 
-# Setup Gemini
-genai.configure(api_key=GEMINI_API_KEY)
-model = genai.GenerativeModel(
-    "gemini-1.5-flash",
-    system_instruction=(
-        "You are Sumit, a friendly and helpful buddy inside a Telegram chat. "
-        "Never mention that you are an AI or Gemini. "
-        "Talk in a casual, human-like, friendly Hinglish style. "
-        "Keep answers clear, supportive, and fun."
-    )
+# ==== LOGGING ====
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
 )
+logger = logging.getLogger(__name__)
 
-# Setup Logging
-logging.basicConfig(level=logging.INFO)
+# ==== GEMINI AI SETUP ====
+genai.configure(api_key=GEMINI_API_KEY)
+model = genai.GenerativeModel("gemini-pro")
 
-# ==== TELEGRAM HANDLERS ====
-async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await update.message.reply_text(
-        "Hey 👋, main Sumit hoon! Mujhse kuch bhi pucho – life, gyaan, ya mazedar baatein. 😊"
-    )
+# ==== FLASK APP ====
+flask_app = Flask(__name__)
 
-async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    user_text = update.message.text
+@flask_app.route("/")
+def home():
+    return "✅ Telegram AI Chatbot is running on port 7860!"
+
+# ==== TELEGRAM BOT HANDLERS ====
+async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    await update.message.reply_text("👋 Hi! I am your AI friend. How can I help you today?")
 
+async def chat(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    user_message = update.message.text
     try:
-        response = model.generate_content(user_text)
-        reply = response.text if response.text else "Hmm, mujhe thoda sochne do 🤔."
+        response = model.generate_content(user_message)
+        reply = response.text if response.text else "⚠️ I couldn’t generate a reply."
     except Exception as e:
-        reply = "Arre, thoda error aaya hai 😅. Thodi der baad try karo."
-
+        logger.error(e)
+        reply = "❌ Something went wrong while generating response."
     await update.message.reply_text(reply)
 
-# ==== FLASK APP FOR STATUS ====
-app = Flask(__name__)
-
-@app.route("/")
-def home():
-    return "✅ Sumit Telegram Bot is running on Hugging Face Space!"
-
-@app.route("/healthz")
-def health():
-    return {"status": "ok", "bot": "Sumit is running"}
-
-def run_flask():
-    # Bind to 0.0.0.0 so HF Spaces can access it
-    app.run(host="0.0.0.0", port=7860)
+# ==== TELEGRAM BOT RUNNER ====
+def run_telegram_bot():
+    app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
+    app.add_handler(CommandHandler("start", start))
+    app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, chat))
+    app.run_polling()
 
 # ==== MAIN ====
-def main():
-    # Run Flask in a separate thread
-    threading.Thread(target=run_flask, daemon=True).start()
-
-    # Start Telegram bot
-    tg_app = Application.builder().token(TELEGRAM_TOKEN).build()
-    tg_app.add_handler(CommandHandler("start", start))
-    tg_app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_message))
-
-    print("✅ Sumit bot is running...")
-    tg_app.run_polling()
-
 if __name__ == "__main__":
-    main()
+    import threading
+    
+    # Run Telegram bot in a separate thread
+    threading.Thread(target=run_telegram_bot, daemon=True).start()
+
+    # Run Flask server on 0.0.0.0:7860
+    flask_app.run(host="0.0.0.0", port=7860)