Update main.py

main.py

@@ -4,16 +4,28 @@ import asyncio
 from datetime import datetime
 from telegram import Update
 from telegram.ext import ApplicationBuilder, ContextTypes, CommandHandler, MessageHandler, filters
+from telegram.request import HTTPXRequest # Importante para consertar o erro de rede
 
-# Import Agent and ReminderManager
+# Tenta carregar variáveis locais (útil se você testar no PC)
 try:
-    from .core import TelegramJadeAgent
-    from .tools import ReminderManager
+    from dotenv import load_dotenv
+    load_dotenv()
 except ImportError:
+    pass
+
+# --- IMPORTS CORRIGIDOS (SEM PONTO) ---
+# Se ainda der erro no core.py, lembre de remover o ponto lá também!
+try:
+    from core import TelegramJadeAgent
+    from tools import ReminderManager
+except ImportError:
+    # Fallback caso a estrutura de pastas mude
+    import sys
+    sys.path.append(os.path.dirname(os.path.abspath(__file__)))
     from core import TelegramJadeAgent
     from tools import ReminderManager
 
-# Logging
+# Logging setup
 logging.basicConfig(
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
     level=logging.INFO
@@ -21,18 +33,21 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 # Global instances
+# Inicializa o gerenciador de lembretes e o agente
 reminder_manager = ReminderManager()
 agent = TelegramJadeAgent(reminder_manager=reminder_manager)
 
 async def start(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    await context.bot.send_message(chat_id=update.effective_chat.id, text="Hi! I'm Jade, your 24h assistant. How can I help you?")
+    await context.bot.send_message(chat_id=update.effective_chat.id, text="Oi! Eu sou a Jade. Como posso ajudar você hoje?")
 
 async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
     chat_id = update.effective_chat.id
     user_text = update.message.text
     
-    # 1. Get response from Agent
-    # agent.chat is blocking (sync), so we wrap it.
+    # Envia "Digitando..." para o usuário saber que estamos processando
+    await context.bot.send_chat_action(chat_id=chat_id, action="typing")
+    
+    # Chama o agente em uma thread separada para não travar o bot
     response = await asyncio.to_thread(agent.chat, chat_id, user_text)
     
     await context.bot.send_message(chat_id=chat_id, text=response)
@@ -46,7 +61,14 @@ async def send_reminder_job(context: ContextTypes.DEFAULT_TYPE):
     
     # Check if reminder is still valid (not deleted)
     current_reminders = reminder_manager.get_due_reminders()
-    if not any(r["id"] == reminder_id for r in current_reminders):
+    # Verifica se o ID ainda existe na lista de pendentes
+    exists = False
+    for r in current_reminders: # Recarrega para ter certeza
+        if r["id"] == reminder_id:
+            exists = True
+            break
+            
+    if not exists:
         logger.info(f"Reminder {reminder_id} skipped (deleted or sent).")
         return
 
@@ -65,7 +87,10 @@ async def check_reminders_loop(context: ContextTypes.DEFAULT_TYPE):
     Background task running every X seconds to check for new reminders in DB
     and schedule them in the JobQueue.
     """
-    pending = reminder_manager.get_due_reminders()
+    # Recarrega lembretes do arquivo
+    reminder_manager._load_reminders()
+    pending = [r for r in reminder_manager.reminders if r["status"] == "pending"]
+    
     job_queue = context.job_queue
     
     # Get current scheduled jobs by name
@@ -73,32 +98,39 @@ async def check_reminders_loop(context: ContextTypes.DEFAULT_TYPE):
     
     for r in pending:
         job_name = f"reminder_{r['id']}"
+        # Se o job ainda não está na fila, agende
         if job_name not in current_jobs:
             try:
                 run_date = datetime.strptime(r["time"], "%Y-%m-%d %H:%M:%S")
                 
-                # If time is in past, schedule for immediate execution
+                # Se for passado, agenda para agora
                 if run_date < datetime.now():
                     run_date = datetime.now()
-                
-                logger.info(f"Scheduling {job_name} for {run_date}")
-                
+                    
+                # Só agenda se não for muito no passado (opcional, mas evita spam de velhos)
+                # Aqui agendamos apenas uma vez
                 job_queue.run_once(
                     send_reminder_job, 
                     when=run_date, 
                     data=r, 
                     name=job_name
                 )
+                logger.info(f"Scheduled {job_name} for {run_date}")
             except Exception as e:
                 logger.error(f"Failed to schedule {job_name}: {e}")
 
 if __name__ == '__main__':
+    # Verifica o Token
     token = os.getenv("TELEGRAM_BOT_TOKEN")
     if not token:
-        print("Error: TELEGRAM_BOT_TOKEN not found.")
+        logger.error("Error: TELEGRAM_BOT_TOKEN not found in environment variables.")
         exit(1)
         
-    application = ApplicationBuilder().token(token).build()
+    # --- CONFIGURAÇÃO DE REDE BLINDADA PARA HUGGING FACE ---
+    # Força HTTP/1.1 e aumenta timeouts para evitar desconexões
+    req = HTTPXRequest(connection_pool_size=8, http_version="1.1", read_timeout=20, write_timeout=20)
+    
+    application = ApplicationBuilder().token(token).request(req).build()
     
     start_handler = CommandHandler('start', start)
     msg_handler = MessageHandler(filters.TEXT & (~filters.COMMAND), handle_message)
@@ -106,9 +138,11 @@ if __name__ == '__main__':
     application.add_handler(start_handler)
     application.add_handler(msg_handler)
     
-    # Add a repeating job to sync reminders (e.g. every 10 seconds)
+    # Verifica lembretes a cada 10 segundos
     if application.job_queue:
         application.job_queue.run_repeating(check_reminders_loop, interval=10, first=1)
     
     print("Jade Telegram Bot is running...")
-    application.run_polling()
+    
+    # drop_pending_updates=True limpa mensagens velhas ao reiniciar para não travar
+    application.run_polling(drop_pending_updates=True)