Update app.py

app.py

@@ -35,19 +35,20 @@ SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 # 대화 히스토리를 저장할 전역 변수
 conversation_history = []
 
+
 def latex_to_image(latex_string):
     plt.figure(figsize=(10, 1))
     plt.axis('off')
-    plt.text(0.5, 0.5, latex_string, size=20, ha='center', va='center', color='white')
+    plt.text(0.5, 0.5, f'${latex_string}$', size=20, ha='center', va='center', color='white')
     
     buffer = BytesIO()
     plt.savefig(buffer, format='png', bbox_inches='tight', pad_inches=0.1, transparent=True, facecolor='black')
     buffer.seek(0)
     
-    image_base64 = base64.b64encode(buffer.getvalue()).decode()
     plt.close()
     
-    return image_base64
+    return buffer
+    
 
 def process_and_convert_latex(text):
     # 단일 $ 또는 이중 $$ 로 둘러싸인 LaTeX 수식을 찾습니다.
@@ -161,24 +162,26 @@ class MyClient(discord.Client):
 
     async def send_message_with_latex(self, channel, message):
         try:
-            # 텍스트와 LaTeX 수식 분리
+        # 텍스트와 LaTeX 수식 분리
             text_parts = re.split(r'(\$\$.*?\$\$|\$.*?\$)', message, flags=re.DOTALL)
         
             for part in text_parts:
                 if part.startswith('$'):
-                    # LaTeX 수식 처리 및 이미지로 출력
+                # LaTeX 수식 처리 및 이미지로 출력
                     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'))
+                    image_buffer = latex_to_image(latex_content)
+                    await channel.send(file=discord.File(fp=image_buffer, filename='equation.png'))
                 else:
-                    # 텍스트 출력
+                # 텍스트 출력
                     if part.strip():
                         await self.send_long_message(channel, part.strip())
     
         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: