Update main.py

main.py

@@ -100,17 +100,14 @@ def upload_to_imgbb(image_base64: str):
             logging.error(f"Imgbb API Response: {response.status_code} - {response.text}")
         return None
 
-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
+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_gemini_response(prompt: str, images: list, api_url: str):
     logging.info("Sending request to Gemini API.")