update prompt parameters

app/api/generate.py

@@ -58,8 +58,7 @@ def text_image(model: Request, request: TextImage, db: Annotated[Session, Depend
         raise HTTPException(status_code=403, detail="Forbidden")
 
     generator = torch.manual_seed(request.seed)
-    prompt = create_prompt(request.prompt, medium=request.medium, style=request.style,
-                           additional_details=request.additional_details, lightning=request.lightning)
+    prompt = create_prompt(request.prompt, medium=request.medium, style=request.style, artist=request.artist, website=request.website, resolution=request.resolution, additional_details=request.additional_details, color=request.color, lightning=request.lightning)
 
     crud.create_prompt(db=db, user_id=current_user.user_id, prompt=prompt)
 
@@ -75,8 +74,7 @@ def image_image(model: Request, request: ImageImage, db: Annotated[Session, Depe
         raise HTTPException(status_code=403, detail="Forbidden")
 
     generator = torch.manual_seed(request.seed)
-    prompt = create_prompt(request.prompt, medium=request.medium, style=request.style,
-                           additional_details=request.additional_details, lightning=request.lightning)
+    prompt = create_prompt(request.prompt, medium=request.medium, style=request.style, artist=request.artist, website=request.website, resolution=request.resolution, additional_details=request.additional_details, color=request.color, lightning=request.lightning)
     image = decode_image(request.image)
 
     crud.create_prompt(db=db, user_id=current_user.user_id, prompt=prompt)
@@ -142,8 +140,7 @@ def image_variations(model: Request, request: ImageVariations, db: Annotated[Ses
     if not current_user.is_active:
         raise HTTPException(status_code=403, detail="Forbidden")
 
-    # prompt = create_prompt(request.prompt, medium=request.medium, style=request.style,
-    #                        additional_details=request.additional_details, lightning=request.lightning)
+    # prompt = create_prompt(request.prompt, medium=request.medium, style=request.style, artist=request.artist, website=request.website, resolution=request.resolution, additional_details=request.additional_details, color=request.color, lightning=request.lightning)
     # image = decode_image(request.image)
     # image.resize((512, 512))
 

app/core/schemas.py

@@ -47,8 +47,12 @@ class Generate(BaseModel):
     negative_prompt : str = "ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face"
     medium: str | None = None
     style: str | None = None
-    lightning: str | None = None
+    artist: str | None = None
+    website: str | None = None
+    resolution: str | None = None
     additional_details: str | None = None
+    color: str | None = None
+    lightning: str | None = None
 
 
 class TextImage(Generate):

app/main.py

@@ -6,16 +6,14 @@ from fastapi.security import OAuth2PasswordRequestForm
 from fastapi import APIRouter, FastAPI, HTTPException, Depends
 
 import torch
-from ip_adapter import IPAdapterXL
+# from ip_adapter import IPAdapterXL
 from transformers import AutoModelForImageSegmentation
 from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler, StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler, StableDiffusionXLPipeline
 
-from app.db import get_db, init_db
-from app.sql import models
 from app.api import user
 from app.api import prompt
-from app.sql.database import engine
 from app.api import generate
+from app.db import get_db, init_db
 from app.config import ACCESS_TOKEN_EXPIRE_MINUTES, DEVICE
 from app.security import authenticate_user, create_access_token, timedelta