Update app.py

app.py

@@ -89,7 +89,6 @@ class MyClient(discord.Client):
         self.is_processing = True
         try:
             if isinstance(message.channel, discord.TextChannel):
-                # 새로운 스레드 생성
                 thread = await message.channel.create_thread(name=f"질문: {message.author.name}", message=message)
                 if self.is_math_question(message.content):
                     text_response = await self.handle_math_question(message.content)
@@ -99,6 +98,9 @@ class MyClient(discord.Client):
                     await self.send_message_with_latex(thread, response)
             else:
                 logging.warning("Message is not in a TextChannel.")
+        except Exception as e:
+            logging.error(f"Error in on_message: {type(e).__name__}: {str(e)}")
+            await message.channel.send("An error occurred while processing the message.")
         finally:
             self.is_processing = False
 
@@ -127,8 +129,8 @@ class MyClient(discord.Client):
 
             combined_response = f"수학 선생님 답변: ```{cohere_result}```"
 
-        except HTTPError as e:
-            logging.error(f"Hugging Face API error: {e}")
+        except Exception as e:
+            logging.error(f"Error in handle_math_question: {type(e).__name__}: {str(e)}")
             combined_response = "An error occurred while processing the request."
 
         return combined_response
@@ -154,8 +156,8 @@ class MyClient(discord.Client):
                 messages, max_tokens=1000, stream=True, temperature=0.7, top_p=0.85))
             full_response = ''.join([part.choices[0].delta.content for part in response if part.choices and part.choices[0].delta and part.choices[0].delta.content])
             conversation_history.append({"role": "assistant", "content": full_response})
-        except HTTPError as e:
-            logging.error(f"Hugging Face API error: {e}")
+        except Exception as e:
+            logging.error(f"Error in generate_response: {type(e).__name__}: {str(e)}")
             full_response = "An error occurred while generating the response."
 
         return f"{user_mention}, {full_response}"
@@ -192,20 +194,24 @@ class MyClient(discord.Client):
     def switch_client(self):
         if self.hf_client == hf_client_primary:
             self.hf_client = hf_client_secondary
-            logging.info("Switched to secondary client.")
+            logging.info("Switched to secondary client (CohereForAI/aya-23-35B).")
         else:
             self.hf_client = hf_client_primary
-            logging.info("Switched back to primary client.")
+            logging.info("Switched back to primary client (CohereForAI/c4ai-command-r-plus).")
 
     async def retry_request(self, func, retries=5, delay=2):
         for i in range(retries):
             try:
                 return await func()
-            except HTTPError as e:
-                if e.response.status_code == 503 and i < retries - 1:
+            except Exception as e:
+                logging.error(f"Error encountered: {type(e).__name__}: {str(e)}")
+                if isinstance(e, HTTPError) and e.response.status_code == 503:
                     logging.warning(f"503 error encountered. Retrying in {delay} seconds...")
                     self.switch_client()  # 클라이언트 전환
                     await asyncio.sleep(delay)
+                elif i < retries - 1:
+                    logging.warning(f"Error occurred. Retrying in {delay} seconds...")
+                    await asyncio.sleep(delay)
                 else:
                     raise