chore: enhance OpenAI integration with connection test endpoint, improved error handling, and validate API key presence

app/page.tsx

@@ -32,7 +32,7 @@ interface ImageGenerationResponse {
 
 const DEFAULT_PROMPT = `這是我創作的角色，叫做狐狸貓，請完全保留圖片中的狐狸貓外觀，包括：
 五官大小、頭身比例、身體各部位大小、耳朵大小及陰影角度。
-注意，狐狸貓角色「沒有」筆觸描邊、圖片上不要有任何文字、透明背景
+注意，狐狸貓角色「沒有」筆觸描邊、圖片上不要有任何文字
 請直接給我圖片，不需要回復任何文字`;
 
 export default function Home() {

backend/app/api/v1/endpoints/images.py

@@ -32,11 +32,11 @@ async def generate_image(request: ImageGenerationRequest):
             )
 
         return ImageGenerationResponse(
-            success=True, 
-            message=result["message"], 
+            success=True,
+            message=result["message"],
             filename=result["filename"],
             filenames=result.get("filenames", []),
-            count=result.get("count", 0)
+            count=result.get("count", 0),
         )
 
     except HTTPException:

backend/app/core/config.py

@@ -6,6 +6,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 
+
 class Settings(BaseSettings):
     """
     Application settings configuration

backend/app/main.py

@@ -3,6 +3,7 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.middleware.trustedhost import TrustedHostMiddleware
 import logging
+import os
 
 from .core.config import settings
 from .api.v1.router import api_router
@@ -45,8 +46,8 @@ def create_app() -> FastAPI:
             "http://localhost:8000",
             "http://127.0.0.1:8000",
             "https://*.hf.space",
-            "https://*.huggingface.co", 
-            "*"  # Allow all origins for Hugging Face deployment
+            "https://*.huggingface.co",
+            "*",  # Allow all origins for Hugging Face deployment
         ],
         allow_credentials=True,
         allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
@@ -69,5 +70,34 @@ app = create_app()
 async def health_check():
     return {"status": "healthy", "message": "Backend is running"}
 
+
+# OpenAI connection test endpoint
+@app.get("/test-openai")
+async def test_openai():
+    """Test OpenAI connection without making actual API calls"""
+    try:
+        from .core.config import settings
+
+        if not settings.OPENAI_API_KEY:
+            return {"status": "error", "message": "OPENAI_API_KEY not configured"}
+
+        api_key_preview = (
+            settings.OPENAI_API_KEY[:10] + "..."
+            if len(settings.OPENAI_API_KEY) > 10
+            else "Too short"
+        )
+
+        return {
+            "status": "ok",
+            "message": "OpenAI API key is configured",
+            "api_key_preview": api_key_preview,
+            "environment": (
+                "production" if os.getenv("NODE_ENV") == "production" else "development"
+            ),
+        }
+    except Exception as e:
+        return {"status": "error", "message": f"OpenAI configuration error: {str(e)}"}
+
+
 # Include API router
 app.include_router(api_router, prefix=settings.API_V1_STR)

backend/app/services/image_service.py

@@ -14,7 +14,24 @@ class ImageGenerationService:
     """Service for handling OpenAI image generation"""
 
     def __init__(self):
-        self.client = OpenAI(api_key=settings.OPENAI_API_KEY)
+        # Validate API key exists
+        if not settings.OPENAI_API_KEY:
+            logger.error("OPENAI_API_KEY not found in environment variables")
+            raise ValueError("OPENAI_API_KEY is required")
+
+        # Log API key status (first few characters only for security)
+        api_key_preview = (
+            settings.OPENAI_API_KEY[:10] + "..."
+            if settings.OPENAI_API_KEY
+            else "Not set"
+        )
+        logger.info(f"OpenAI API Key status: {api_key_preview}")
+
+        self.client = OpenAI(
+            api_key=settings.OPENAI_API_KEY,
+            timeout=60.0,  # Increase timeout for Hugging Face environment
+            max_retries=2,  # Reduce retries to fail faster
+        )
         self.output_dir = "generated_images"
         self._ensure_output_directory()
 
@@ -48,7 +65,6 @@ class ImageGenerationService:
             logger.info(f"Generating {n} image(s) with prompt: {prompt}")
 
             if reference_image:
-                logging.info(f"Reference image: {settings.OPENAI_API_KEY}")
                 # Use the newer responses API with image generation tools for reference images
                 logger.info("Using reference image with responses API")
 
@@ -112,7 +128,28 @@ class ImageGenerationService:
                         logger.info(f"Image {i+1} saved successfully: {filename}")
 
                     except Exception as e:
-                        logger.warning(f"Failed to generate image {i+1}: {str(e)}")
+                        error_msg = str(e)
+                        logger.warning(f"Failed to generate image {i+1}: {error_msg}")
+
+                        # More specific error handling for network issues
+                        if (
+                            "Connection error" in error_msg
+                            or "timeout" in error_msg.lower()
+                        ):
+                            logger.error(
+                                f"Network connectivity issue detected: {error_msg}"
+                            )
+                            logger.error(
+                                "This might be due to Hugging Face network restrictions"
+                            )
+                        elif (
+                            "api_key" in error_msg.lower()
+                            or "unauthorized" in error_msg.lower()
+                        ):
+                            logger.error(f"API key issue detected: {error_msg}")
+                        elif "rate limit" in error_msg.lower():
+                            logger.error(f"Rate limit issue detected: {error_msg}")
+
                         continue
 
                 if not generated_filenames: