Update main.py

main.py

@@ -103,7 +103,6 @@ def generate_and_upload_image(prompt: str) -> str:
         生成圖片的 URL。
     """
     try:
-        # genai_client = genai.Client(api_key=google_api)
         response = genai_client.models.generate_content(
             model="gemini-2.0-flash-exp",
             contents=prompt,
@@ -246,11 +245,10 @@ def handle_message(event):
             # 運行代理人
             response = agent_executor.invoke(agent_input)
             out = response["output"]
-            
+            print('out')
             if 'htts' in out:
-                img_tmp = out.split('https')[1]
-                print('img_tmp',img_tmp)
-                image_url = img_tmp.split('png')[0]
+                img_tmp = 'https'+out.split('https')[1]
+                image_url = img_tmp.split('png')[0]+'png'
                 print('image_url',image_url)
                 line_bot_api.push_message(
                     event.source.user_id,