Update app/schemas/image.py

app/schemas/image.py

@@ -1,8 +1,5 @@
 from pydantic import BaseModel
 from typing import Optional, Dict, Any
-from fastapi import UploadFile, File # Import UploadFile and File
-
-# --- Request Models ---
 
 class EnhancePromptRequest(BaseModel):
     raw_prompt: str
@@ -11,33 +8,28 @@ class GenerateImageRequest(BaseModel):
     raw_prompt: Optional[str] = None
     enhanced_prompt: Optional[str] = None
 
-# UpdateImageRequest still needs base64 as it receives existing image for modification
 class UpdateImageRequest(BaseModel):
     text_instruction: str
     image_base64: str
 
-# VirtualTryOnRequest now expects UploadFile objects
-# Note: For file uploads, it's often simpler to define them directly in the API endpoint function
-# using `File()` or `Form()` for metadata, rather than in a Pydantic model for the request body.
-# However, for consistency, we'll keep a schema here and handle parsing in the router.
-# But for multipart/form-data, Pydantic models for request *body* are slightly different.
-# The endpoint itself will directly take the UploadFile.
-
-
-# --- Response Models ---
-
 class EnhancePromptResponse(BaseModel):
     raw_prompt: str
     enhanced_prompt: str
 
-# These response models no longer contain image_base64
-# The image itself will be the response body.
 class GenerateImageOnlyResponse(BaseModel):
-    text: Optional[str] = None # Text can still be returned if the model generates it
+    text: Optional[str] = None 
 
 class UpdateImageOnlyResponse(BaseModel):
     text: Optional[str] = None
 
 class VirtualTryOnResponse(BaseModel):
     summary: Dict[str, Any]
-    # try_on_image will be the response body, so no base64 here
+
+class ShoeGenerateRequest(BaseModel):
+    style: str
+    color: Optional[str] = None
+    perspective: Optional[str] = None
+    variations: Optional[int] = 1
+
+class ShoeGenerateResponse(BaseModel):
+    text: Optional[str] = None