Fix app.py

app.py

@@ -42,6 +42,7 @@ def create_prompt_for_image_generation(user_prompt: str) -> str:
 
     Args:
         user_prompt: The user's text prompt to be processed by the language model.
+    Output type: str
     """
     # Prompt parts
     prefix="Generate a detailed and structured FLUX-Schnell-compatible prompt based on the following short description of an image: "
@@ -65,11 +66,11 @@ def create_prompt_for_image_generation(user_prompt: str) -> str:
     The color palette includes rich greens, warm browns for the cabin, and soft gray mist. 
     The perspective is slightly elevated as if viewed from a drone camera at sunrise, 
     capturing golden hour lighting for soft shadows and warm highlights. 
-    The aspect ratio is 16:9 in PNG format with high quality (90), using seed 42 for reproducibility.
+    The aspect ratio is 1:1, using seed 42 for reproducibility.
     '''
     """
     model = HfApiModel(
-    max_tokens=200,
+    max_tokens=512,
     temperature=1.0,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
     custom_role_conversions=None,
@@ -77,13 +78,8 @@ def create_prompt_for_image_generation(user_prompt: str) -> str:
     prompt = prefix + user_prompt + '. ' + postfix 
     try:
         response = model.generate(
-            prompt=prompt, temperature=1., max_tokens=200)
-
-        response = openai.Completion.create(
-            engine=model_name,  # Use 'engine' for older models, 'model' for newer
-            prompt=prompt
-        )
-        return response.choices[0].text.strip()
+            prompt=prompt, temperature=1., max_tokens=512)
+        return response['choices'][0]['text'] 
 
     except Exception as e:
         return f"Error during LLM call: {str(e)}"