Update app.py

app.py

@@ -36,12 +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')
+    plt.text(0.5, 0.5, latex_string, size=20, ha='center', va='center')
     
     buffer = BytesIO()
     plt.savefig(buffer, format='png', bbox_inches='tight', pad_inches=0.1, transparent=True)
@@ -51,6 +48,8 @@ def latex_to_image(latex_string):
     plt.close()
     
     return image_base64
+    
+
 
 def process_and_convert_latex(text):
     # 단일 $ 또는 이중 $$ 로 둘러싸인 LaTeX 수식을 찾습니다.
@@ -158,21 +157,28 @@ class MyClient(discord.Client):
         return f"{user_mention}, {full_response}"
 
     async def send_message_with_latex(self, channel, message):
-        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.")
+    try:
+        # 텍스트와 LaTeX 수식 분리
+        text_parts = re.split(r'(\$\$.*?\$\$|\$.*?\$)', message, flags=re.DOTALL)
+        
+        # 텍스트 먼저 출력
+        for part in text_parts:
+            if not part.startswith('$'):
+                if part.strip():
+                    await self.send_long_message(channel, part.strip())
+        
+        # LaTeX 수식 처리 및 이미지로 출력
+        for part in text_parts:
+            if part.startswith('$'):
+                latex_content = part.strip('$')
+                image_base64 = latex_to_image(latex_content)
+                image_binary = base64.b64decode(image_base64)
+                await channel.send(file=discord.File(BytesIO(image_binary), 'equation.png'))
+    
+    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: