Update main.py

main.py

@@ -53,7 +53,7 @@ Enhance the prompt by including:
 - Call-to-Action: include a clear, concise CTA in the poster text (e.g., "Shop Now", "Learn More")
 
 Business Description:
-{raw_prompt}
+{Raw_Prompt}
 
 Enhanced Poster Prompt:
 ''')
@@ -64,11 +64,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 +90,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 +133,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.")