fix: Improve Discord image URL retrieval and handle message attachments

fix: replace the base_url in webui
Feiyuyu0503 · Jul 13, 2024 · 06ea8b2 · 06ea8b2
1 parent 20b5fb4
commit 06ea8b2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Bots/discordClient.py b/Bots/discordClient.py
@@ -70,8 +70,8 @@ async def on_message(self, message):
                         else:
                             raise Exception("No message content found")
                     else:
-                        await asyncio.sleep(5)
-                        msg = message.content
+                        await asyncio.sleep(10)
+                        msg = message.attachments[0].url if message.attachments else None
                         self.pending_responses[reply_to_msg_id] = (response_event, msg)
                 except Exception as e:
                     logger.error(f"Discord client handle response error: {e}")

diff --git a/webui/gradio_ui.py b/webui/gradio_ui.py
@@ -7,7 +7,7 @@
 current_task = None
 
 async def get_image_url(text,api_key,request:gr.Request):
-    base_url = request.headers.get("host")
+    base_url = request.headers.get("host").replace("localhost", "127.0.0.1")
     scheme = request.headers.get("x-forwarded-proto") or "http"
     global current_task
     # 向 FastAPI 后端发送 POST 请求，并获取图片 URL