Update bot_telegram.py

bot_telegram.py

@@ -105,53 +105,57 @@ class TelegramBot:
             await update.message.reply_text("❌ Wrong password. Try again.")
 
     async def chat_with_ai(self, update: Update, context: CallbackContext):
-    """Handles messages and sends them to the AI API."""
-    user_id = update.message.from_user.id
+        """Handles messages and sends them to the AI API."""
+        user_id = update.message.from_user.id
 
-    if user_id not in AUTHENTICATED_USERS:
-        await update.message.reply_text("❌ You are not authenticated. Please enter the password first.")
-        return
+        if user_id not in AUTHENTICATED_USERS:
+            await update.message.reply_text("❌ You are not authenticated. Please enter the password first.")
+            return
 
-    if not self.auth_token:
-        self.authenticate()
+        if not self.auth_token:
+            self.authenticate()
 
-    if not self.auth_token:
-        await update.message.reply_text("Authentication failed. Please try again later.")
-        return
+        if not self.auth_token:
+            await update.message.reply_text("Authentication failed. Please try again later.")
+            return
 
-    user_message = update.message.text
+        user_message = update.message.text
 
-    headers = {
-        "Authorization": f"Bearer {self.auth_token}",
-        "Content-Type": "application/x-www-form-urlencoded",  # Always use form-data
-        "accept": "application/json"
-    }
+        headers = {
+            "Authorization": f"Bearer {self.auth_token}",
+            "accept": "application/json"
+        }
 
-    form_payload = {"question": user_message}  # Only use form-data
+        json_payload = {"question": user_message}
+        form_payload = {"question": user_message}
 
-    try:
-        response = requests.post(self.ai_url, headers=headers, data=form_payload)
+        try:
+            logging.info(f"Sending payload as JSON: {json_payload}")
+            response = requests.post(self.ai_url, headers={**headers, "Content-Type": "application/json"}, json=json_payload)
 
-        logging.info(f"Response status: {response.status_code}")
-        logging.info(f"Response content: {response.text}")
+            if response.status_code == 422:
+                logging.warning("JSON format rejected. Retrying with form-data...")
+                response = requests.post(self.ai_url, headers={**headers, "Content-Type": "application/x-www-form-urlencoded"}, data=form_payload)
 
-        if response.status_code == 200:
-            bot_reply = response.json().get("answer", "I didn't understand that.")
-        elif response.status_code == 401:
-            logging.warning("Authorization expired. Re-authenticating...")
-            self.authenticate()
-            await self.chat_with_ai(update, context)
-            return
-        elif response.status_code == 422:
-            bot_reply = "Error: The API rejected the request. Check payload format."
-        else:
-            bot_reply = f"Error: {response.status_code} - {response.text}"
+            logging.info(f"Response status: {response.status_code}")
+            logging.info(f"Response content: {response.text}")
 
-    except Exception as e:
-        bot_reply = f"Connection error: {e}"
+            if response.status_code == 200:
+                bot_reply = response.json().get("answer", "I didn't understand that.")
+            elif response.status_code == 401:
+                logging.warning("Authorization expired. Re-authenticating...")
+                self.authenticate()
+                await self.chat_with_ai(update, context)
+                return
+            elif response.status_code == 422:
+                bot_reply = "Error: The API rejected the request. Check payload format."
+            else:
+                bot_reply = f"Error: {response.status_code} - {response.text}"
 
-    await update.message.reply_text(bot_reply)
+        except Exception as e:
+            bot_reply = f"Connection error: {e}"
 
+        await update.message.reply_text(bot_reply)
 
     def setup_handlers(self):
         """Set up Telegram command and message handlers."""