Update app.py

app.py

@@ -71,57 +71,43 @@ class MyClient(discord.Client):
         super().__init__(*args, **kwargs)
         self.is_processing = False
         self.math_pipe = math_pipe
-        self.hf_client = hf_client_primary  # 초기 클라이언트 설정
-
-    async def on_ready(self):
-        logging.info(f'{self.user}로 로그인되었습니다!')
-        subprocess.Popen(["python", "web.py"])
-        logging.info("Web.py server has been started.")
-
-    async def on_message(self, message):
-        if message.author == self.user:
-            return
-        if not self.is_message_in_specific_channel(message):
-            return
-        if self.is_processing:
-            return
-
-        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)
-                    await self.send_message_with_latex(thread, text_response)
-                else:
-                    response = await self.generate_response(message)
-                    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
+        self.current_client = "primary"
+        self.hf_client = hf_client_primary
 
-    def is_message_in_specific_channel(self, message):
-        return isinstance(message.channel, discord.TextChannel) and (message.channel.id == SPECIFIC_CHANNEL_ID or (
-            isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
-        ))
+    def switch_client(self):
+        if self.current_client == "primary":
+            self.hf_client = hf_client_secondary
+            self.current_client = "secondary"
+            logging.info("Switched to secondary client (CohereForAI/aya-23-35B).")
+        else:
+            self.hf_client = hf_client_primary
+            self.current_client = "primary"
+            logging.info("Switched back to primary client (CohereForAI/c4ai-command-r-plus).")
 
-    def is_math_question(self, content):
-        return bool(re.search(r'\b(solve|equation|calculate|math)\b', content, re.IGNORECASE))
+    async def retry_request(self, func, retries=5, delay=2):
+        for i in range(retries):
+            try:
+                return await func()
+            except Exception as e:
+                logging.error(f"Attempt {i+1}/{retries}: Error encountered: {type(e).__name__}: {str(e)}")
+                if isinstance(e, HTTPError) and getattr(e.response, 'status_code', None) == 503:
+                    logging.warning(f"503 error encountered. Switching client and retrying in {delay} seconds...")
+                    self.switch_client()
+                elif i < retries - 1:
+                    logging.warning(f"Error occurred. Retrying in {delay} seconds...")
+                await asyncio.sleep(delay)
+        
+        logging.error(f"All {retries} attempts failed.")
+        raise Exception("Max retries reached")
 
     async def handle_math_question(self, question):
         loop = asyncio.get_event_loop()
         
-        # AI-MO/NuminaMath-7B-TIR 모델에게 수학 문제를 풀도록 요청
         math_response_future = loop.run_in_executor(None, lambda: self.math_pipe(question, max_new_tokens=2000))
         math_response = await math_response_future
         math_result = math_response[0]['generated_text']
         
         try:
-            # Cohere 모델에게 AI-MO/NuminaMath-7B-TIR 모델의 결과를 번역하도록 요청
             cohere_response = await self.retry_request(lambda: self.hf_client.chat_completion(
                 [{"role": "system", "content": "다음 텍스트를 한글로 번역하십시오: "}, {"role": "user", "content": math_result}], max_tokens=1000))
         
@@ -135,6 +121,9 @@ class MyClient(discord.Client):
 
         return combined_response
 
+
+
+
     async def generate_response(self, message):
         global conversation_history
         user_input = message.content