Update app.py

app.py

@@ -72,9 +72,9 @@ class MyClient(discord.Client):
             isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
         )
 
-
-async def generate_image_response(message, image_data):
+async def generate_response(message):
     global conversation_history
+    user_input = message.content
     user_mention = message.author.mention
     
     system_message = f"""
@@ -85,7 +85,7 @@ async def generate_image_response(message, image_data):
     {user_mention}, DISCORD에서 사용자들의 질문에 답하는 어시스턴트입니다.
     """
 
-    conversation_history.append({"role": "user", "content": "이미지를 분석해 주세요."})
+    conversation_history.append({"role": "user", "content": user_input})
     logging.debug(f'Conversation history updated: {conversation_history}')
 
     response = claude_client.messages.create(
@@ -93,8 +93,7 @@ async def generate_image_response(message, image_data):
         max_tokens=1000,
         temperature=0.7,
         system=system_message,
-        messages=conversation_history,
-        files=[("image.png", io.BytesIO(image_data))]  # 수정
+        messages=conversation_history
     )
 
     full_response_text = response.content[0].text
@@ -102,8 +101,6 @@ async def generate_image_response(message, image_data):
     
     conversation_history.append({"role": "assistant", "content": full_response_text})
     return f"{user_mention}, {full_response_text}"
-    
-
 
 async def generate_image_response(message, image_data):
     global conversation_history
@@ -126,7 +123,7 @@ async def generate_image_response(message, image_data):
         temperature=0.7,
         system=system_message,
         messages=conversation_history,
-        attachments=[anthropic.Attachment(io.BytesIO(image_data), "image.png")]
+        files=[("image.png", io.BytesIO(image_data))]
     )
 
     full_response_text = response.content[0].text
@@ -156,7 +153,7 @@ async def generate_data_response(message, data, filename):
         temperature=0.7,
         system=system_message,
         messages=conversation_history,
-        attachments=[anthropic.Attachment(io.BytesIO(data), filename)]
+        files=[("data.csv", io.BytesIO(data))]
     )
 
     full_response_text = response.content[0].text
@@ -176,5 +173,4 @@ async def send_long_message(channel, message):
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
-    discord_client.run(os.getenv('DISCORD_TOKEN'))
-    
+    discord_client.run(os.getenv('DISCORD_TOKEN'))