Update bot_telegram.py

bot_telegram.py

@@ -105,57 +105,53 @@ 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}",
-            "accept": "application/json"
-        }
+    headers = {
+        "Authorization": f"Bearer {self.auth_token}",
+        "Content-Type": "application/x-www-form-urlencoded",  # Always use form-data
+        "accept": "application/json"
+    }
 
-        json_payload = {"question": user_message}
-        form_payload = {"question": user_message}
+    form_payload = {"question": user_message}  # Only use form-data
 
-        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)
+    try:
+        response = requests.post(self.ai_url, headers=headers, data=form_payload)
 
-            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)
+        logging.info(f"Response status: {response.status_code}")
+        logging.info(f"Response content: {response.text}")
 
-            logging.info(f"Response status: {response.status_code}")
-            logging.info(f"Response content: {response.text}")
+        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}"
 
-            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}"
+    except Exception as e:
+        bot_reply = f"Connection error: {e}"
 
-        except Exception as e:
-            bot_reply = f"Connection error: {e}"
+    await update.message.reply_text(bot_reply)
 
-        await update.message.reply_text(bot_reply)
 
     def setup_handlers(self):
         """Set up Telegram command and message handlers."""