Update main.py

main.py

@@ -287,14 +287,14 @@ async def generate_story_endpoint(image_file: UploadFile = File(...), language:
 
     # Step 2: Construct the prompt text for the language model
     # This prompt instructs the model on what to write and incorporates the caption.
-    prompt_text = f"Write an attractive story of around 300 words about {story_theme}. Incorporate the following details from an image description into the story: {caption}\n\nStory:"
+    prompt_text = f"Write a detailed story that is approximately 300 words long. Ensure the story has a clear beginning, middle, and end about {story_theme}. Incorporate the following details from an image description into the story: {caption}\n\nStory:"
 
     # Step 3: Generate the story using the local language model (Qwen 0.5B via llama.cpp)
     try:
         # Call the Qwen 0.5B story generation function
         story = generate_story_qwen_0_5b( # <--- Use the updated function name
             prompt_text,
-            max_new_tokens=300, # Request ~300 new tokens
+            max_new_tokens=350, # Request ~300 new tokens
             temperature=0.7,    # Sampling parameters
             top_p=0.9,
             top_k=50            # Note: top_k may not be directly used by llama_cpp.create_chat_completion