Update main.py

main.py

@@ -100,14 +100,17 @@ def upload_to_imgbb(image_base64: str):
             logging.error(f"Imgbb API Response: {response.status_code} - {response.text}")
         return None
 
-if image_binary:
-        image = PIL.Image.open(io.BytesIO(image_binary))
-        # 隨機生成一個檔案名以避免衝突
-        file_name = f"static/{os.urandom(16).hex()}.png"
-        image.save(file_name, format="PNG")
-            
-        image_url = os.path.join(os.getenv("HF_SPACE"), file_name)
-return image_url
+def get_base64_from_url(image_url: str):
+    """
+    從 URL 下載圖片並轉換為 Base64 編碼。
+    """
+    try:
+        response = requests.get(image_url)
+        response.raise_for_status()
+        return base64.b64encode(response.content).decode('utf-8')
+    except requests.exceptions.RequestException as e:
+        logging.error(f"Error fetching image from URL: {image_url}, error: {e}")
+        return None
 
 def get_gemini_response(prompt: str, images: list, api_url: str):
     logging.info("Sending request to Gemini API.")