Add error handling and logging to webhook endpoints

telegram_handlers.py

@@ -5,60 +5,74 @@ from ai_service import get_ai_response
 from database import db_manager
 
 async def telegram_webhook(request: Request):
-    data = await request.json()
-    
-    if "message" in data:
-        telegram_id = data["message"]["chat"]["id"]
-        user_text = data["message"].get("text", "")
+    try:
+        data = await request.json()
+        print(f"Received webhook data: {data}")
         
-        # Extract user info
-        user_info = data["message"]["chat"]
-        username = user_info.get("username")
-        first_name = user_info.get("first_name")
-        last_name = user_info.get("last_name")
-
-        if user_text:
-            # Save user message to database if available
-            if db_manager:
-                db_manager.save_message(telegram_id, user_text, "user")
-                db_manager.create_or_update_user(telegram_id, username, first_name, last_name)
-            
-            # Get the intelligent response with conversation history
-            ai_answer = await get_ai_response(user_text, telegram_id)
-            
-            # Save assistant response to database if available
-            if db_manager:
-                db_manager.save_message(telegram_id, ai_answer, "assistant")
+        if "message" in data:
+            telegram_id = data["message"]["chat"]["id"]
+            user_text = data["message"].get("text", "")
             
-            # Send back to Telegram if TELEGRAM_URL is available
-            if TELEGRAM_URL:
-                async with httpx.AsyncClient(verify=False) as client:
-                    await client.post(
-                        TELEGRAM_URL,
-                        headers={"Host": "api.telegram.org"},
-                        json={
-                            "chat_id": telegram_id,
-                            "text": ai_answer,
-                            "parse_mode": "Markdown"
-                        }
-                    )
+            # Extract user info
+            user_info = data["message"]["chat"]
+            username = user_info.get("username")
+            first_name = user_info.get("first_name")
+            last_name = user_info.get("last_name")
+
+            if user_text:
+                # Save user message to database if available
+                if db_manager:
+                    db_manager.save_message(telegram_id, user_text, "user")
+                    db_manager.create_or_update_user(telegram_id, username, first_name, last_name)
+                
+                # Get the intelligent response with conversation history
+                ai_answer = await get_ai_response(user_text, telegram_id)
+                
+                # Save assistant response to database if available
+                if db_manager:
+                    db_manager.save_message(telegram_id, ai_answer, "assistant")
                 
-    return {"status": "ok"}
+                # Send back to Telegram if TELEGRAM_URL is available
+                if TELEGRAM_URL:
+                    async with httpx.AsyncClient(verify=False) as client:
+                        await client.post(
+                            TELEGRAM_URL,
+                            headers={"Host": "api.telegram.org"},
+                            json={
+                                "chat_id": telegram_id,
+                                "text": ai_answer,
+                                "parse_mode": "Markdown"
+                            }
+                        )
+                    
+        return {"status": "ok"}
+    except Exception as e:
+        print(f"Error in webhook: {str(e)}")
+        return {"status": "error", "message": str(e)}
 
 async def test_webhook(request: Request):
-    data = await request.json()
-    telegram_id = data["message"]["chat"]["id"]
-    user_text = data["message"]["text"]
-    
-    # Save user message to database if available
-    if db_manager:
-        db_manager.save_message(telegram_id, user_text, "user")
-    
-    # Get response with conversation history
-    response = await get_ai_response(user_text, telegram_id)
-    
-    # Save assistant response to database if available
-    if db_manager:
-        db_manager.save_message(telegram_id, response, "assistant")
-    
-    return {"response": response}
+    try:
+        data = await request.json()
+        print(f"Received test webhook data: {data}")
+        
+        if "message" not in data:
+            return {"error": "Missing 'message' field in request"}
+            
+        telegram_id = data["message"]["chat"]["id"]
+        user_text = data["message"]["text"]
+        
+        # Save user message to database if available
+        if db_manager:
+            db_manager.save_message(telegram_id, user_text, "user")
+        
+        # Get response with conversation history
+        response = await get_ai_response(user_text, telegram_id)
+        
+        # Save assistant response to database if available
+        if db_manager:
+            db_manager.save_message(telegram_id, response, "assistant")
+        
+        return {"response": response}
+    except Exception as e:
+        print(f"Error in test webhook: {str(e)}")
+        return {"error": str(e)}