Update app.py

app.py

@@ -36,6 +36,9 @@ SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 conversation_history = []
 
 def latex_to_image(latex_string):
+    # $$ 기호 제거
+    latex_string = latex_string.strip('$')
+    
     plt.figure(figsize=(10, 1))
     plt.axis('off')
     plt.text(0.5, 0.5, f'${latex_string}$', size=20, ha='center', va='center')
@@ -50,12 +53,18 @@ def latex_to_image(latex_string):
     return image_base64
 
 def process_and_convert_latex(text):
-    latex_pattern = r'\$(.*?)\$'
+    # 단일 $ 또는 이중 $$ 로 둘러싸인 LaTeX 수식을 찾습니다.
+    latex_pattern = r'\$\$(.*?)\$\$|\$(.*?)\$'
     matches = re.findall(latex_pattern, text)
     
-    for match in matches:
-        image_base64 = latex_to_image(match)
-        text = text.replace(f'${match}$', f'<latex_image:{image_base64}>')
+    for double_match, single_match in matches:
+        match = double_match or single_match
+        if match:
+            image_base64 = latex_to_image(match)
+            if double_match:
+                text = text.replace(f'$${match}$$', f'<latex_image:{image_base64}>')
+            else:
+                text = text.replace(f'${match}$', f'<latex_image:{image_base64}>')
     
     return text
 
@@ -149,20 +158,21 @@ class MyClient(discord.Client):
         return f"{user_mention}, {full_response}"
 
     async def send_message_with_latex(self, channel, message):
-        # 텍스트와 LaTeX 수식 분리
-        processed_message = process_and_convert_latex(message)
-        parts = processed_message.split('<latex_image:')
-        
-        for part in parts:
-            if part.startswith('data:image'):
-                # LaTeX 이미지 부분
-                image_data = part.split('>')[0]
-                image_binary = base64.b64decode(image_data)
-                await channel.send(file=discord.File(BytesIO(image_binary), 'equation.png'))
-            else:
-                # 텍스트 부분
-                if part.strip():
-                    await self.send_long_message(channel, part)
+        try:
+            processed_message = process_and_convert_latex(message)
+            parts = processed_message.split('<latex_image:')
+            
+            for part in parts:
+                if part.startswith('data:image'):
+                    image_data = part.split('>')[0]
+                    image_binary = base64.b64decode(image_data)
+                    await channel.send(file=discord.File(BytesIO(image_binary), 'equation.png'))
+                else:
+                    if part.strip():
+                        await self.send_long_message(channel, part)
+        except Exception as e:
+            logging.error(f"Error in send_message_with_latex: {str(e)}")
+            await channel.send("An error occurred while processing the message.")
 
     async def send_long_message(self, channel, message):
         if len(message) <= 2000: