Update app.py

app.py

@@ -6,7 +6,7 @@ from transformers import pipeline
 import asyncio
 import subprocess
 import re
-
+import urllib.parse
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
 
@@ -50,13 +50,16 @@ class MyClient(discord.Client):
         self.is_processing = True
         try:
             if self.is_math_question(message.content):
-                response = await self.handle_math_question(message.content)
+                text_response, img_url = await self.handle_math_question(message.content)
+                await self.send_long_message(message.channel, text_response)
+                await self.send_long_message(message.channel, img_url)  # 이미지 URL을 별도의 메시지로 전송
             else:
                 response = await self.generate_response(message)
-            await self.send_long_message(message.channel, response)
+                await self.send_long_message(message.channel, response)
         finally:
             self.is_processing = False
 
+
     def is_message_in_specific_channel(self, message):
         return message.channel.id == SPECIFIC_CHANNEL_ID or (
             isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
@@ -65,15 +68,19 @@ class MyClient(discord.Client):
     def is_math_question(self, content):
         return bool(re.search(r'\b(solve|equation|calculate|math)\b', content, re.IGNORECASE))
 
-    async def handle_math_question(self, question):
-        loop = asyncio.get_event_loop()
-        response = await loop.run_in_executor(None, lambda: self.math_pipe([{"role": "user", "content": question}]))
-    # 수학적 수식을 LaTeX 포맷으로 감싸기
-        math_response = response[0]['generated_text']
-    # 디스코드에서 LaTeX 수식으로 보여주기 위한 형식
-        latex_output = f"$$\n{math_response}\n$$"
-        return latex_output
-
+async def handle_math_question(self, question):
+    loop = asyncio.get_event_loop()
+    response = await loop.run_in_executor(None, lambda: self.math_pipe([{"role": "user", "content": question}]))
+    math_response = response[0]['generated_text']
+    
+    # QuickLaTeX API를 이용하여 LaTeX 수식을 이미지로 변환
+    encoded_formula = urllib.parse.quote_plus(math_response)
+    quicklatex_url = f"https://quicklatex.com/latex3.f/png?formula={encoded_formula}"
+    image_response = requests.get(quicklatex_url)
+    image_url = image_response.text.split('\n')[1]  # 응답에서 이미지 URL 추출
+
+    return math_response, image_url
+    
     async def generate_response(self, message):
         global conversation_history
         user_input = message.content