APIきーset

app.py

@@ -1007,12 +1007,14 @@ async def mosaic_face(file: UploadFile = File(...)):
 
 
 
-async def classify_image_llm2(file: UploadFile = File(...)):
+async def classify_image_llm(file: UploadFile = File(...), api_key: str = None):
     image_path = "./temp_image.jpg"
     # アップロードされた画像を保存
     with open(image_path, "wb") as buffer:
         buffer.write(await file.read())
-    danger_level = ObjectDetector().detect_danger_level(image_path)
+    if not api_key:
+        api_key = os.getenv('GEMINI_API_KEY')
+    danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
     return {"danger":danger_level}
 
 
@@ -1026,12 +1028,14 @@ async def classify_image_llm2(file: UploadFile = File(...)):
 
 # LLMを使用して画像を分類するエンドポイント
 @app.post("/classify-image-llm/")
-async def classify_image_llm(file: UploadFile = File(...)):
+async def classify_image_llm(file: UploadFile = File(...), api_key: str = None):
     image_path = "./temp_image.jpg"
     # アップロードされた画像を保存
     with open(image_path, "wb") as buffer:
         buffer.write(await file.read())
-    danger_level = ObjectDetector().detect_danger_level(image_path)
+    if not api_key:
+        api_key = os.getenv('GEMINI_API_KEY')
+    danger_level = ObjectDetector(API_KEY=api_key).detect_danger_level(image_path)
     return {"danger":danger_level}
 
 '''