Update main.py

main.py

@@ -39,23 +39,22 @@ llm = get_llm(LANGCHAIN_API_KEY)
 
 # Define the prompt template for enhancing raw prompts.
 prompt_template = PromptTemplate.from_template('''
-You are a Prompt Enhancement AI Assistant specialized in crafting marketing posters that effectively communicate a brand's message and capture audience attention. Using the provided business description and the user's raw poster concept, generate a detailed, professional prompt optimized for AI-driven poster creation.
-
-Enhance the prompt by including:
-- always write generate image, dont use more text
-- Visual Style: (e.g., flat design, isometric, photorealistic, illustrative)
-- Color Scheme & Branding: (e.g., vibrant brand colors, pastel accents, monochrome palette)
-- Typography: (e.g., bold sans-serif headline, elegant serif subtext)
-- Composition & Layout: (e.g., focal point placement, use of negative space, layered elements)
-- Imagery Elements: (e.g., product shots, icons, background patterns, textures)
-- Lighting & Mood: (e.g., bright, dramatic shadows, soft ambient glow)
-- Contextual Details: (e.g., logo placement, tagline integration, call-to-action placement)
+You are a Prompt Enhancement AI Assistant. Your task is to take the user's raw poster image prompt and convert it into a detailed, professional prompt optimized for generating high-quality AI poster.
+
+Enhance the prompt by including relevant details such as:
+- Camera specifications (e.g., lens type, aperture, focal length)
+- Lighting setup (e.g., natural light, studio lighting, soft shadows)
+- Camera angle (e.g., top-down, macro, isometric, side view)
+- Background style (e.g., plain white, minimalistic, outdoor, studio backdrop)
+- Scene composition (e.g., centered product, depth of field, reflections)
 - Call-to-Action: include a clear, concise CTA in the poster text (e.g., "Shop Now", "Learn More")
 
-Business Description:
+Focus only on *Poster photography* — do not include humans or models.
+
+Raw Prompt:
 {Raw_Prompt}
 
-Enhanced Poster Prompt:
+Enhanced Prompt:
 ''')
 
 # Initialize the Google GenAI client.
@@ -64,11 +63,11 @@ client1 = genai.Client(api_key=GOOGLE_API_KEY)
 # --- Request Models ---
 
 class EnhancePromptRequest(BaseModel):
-    Raw_Prompt: str
+    raw_prompt: str
 
 class GenerateImageRequest(BaseModel):
     # If both are provided, enhanced_prompt takes priority.
-    Raw_Prompt: str = None
+    raw_prompt: str = None
     enhanced_prompt: str = None
 
 class UpdateImageRequest(BaseModel):
@@ -90,7 +89,7 @@ async def root():
 async def enhance_prompt(request: EnhancePromptRequest):
     try:
         # Prepare the prompt using the template.
-        formatted_prompt = prompt_template.invoke({"Raw_Prompt": request.Raw_Prompt})
+        formatted_prompt = prompt_template.invoke({"Raw_Prompt": request.raw_prompt})
         # Call the LLM to enhance the prompt.
         response = llm.invoke(formatted_prompt)
         # Assume the enhanced prompt is in the response.content.
@@ -133,7 +132,7 @@ async def generate_image(request: GenerateImageRequest):
         if request.enhanced_prompt:
             image_prompt = request.enhanced_prompt
         elif request.raw_prompt:
-            image_prompt = request.Raw_Prompt
+            image_prompt = request.raw_prompt
         else:
             raise HTTPException(status_code=400, detail="Either raw_prompt or enhanced_prompt must be provided.")