Update app.py

app.py

@@ -37,13 +37,34 @@ def req_flux(token, prompt_value, aspect_ratio="1:1", output_format="webp", num_
         print(f"Error making request: {e}")
         return None
 
-def generate_optimized_prompt(api_key, api_base, system_prompt, user_input):
+def generate_optimized_prompt(api_key, api_base, user_input):
     client = OpenAI(api_key=api_key, base_url=api_base)
 
+    prompt = (
+        f"As a Stable Diffusion prompt expert, your task is to directly modify the following sentence: '{user_input}'"
+        "into a proper Stable Diffusion prompt in English, formatted according to the guidelines provided: "
+        "1. Create prompts using common vocabulary, arranged by importance and separated by commas. Avoid using '-' or '.', "
+        "but spaces and natural language are acceptable. Avoid repeating words. "
+        "2. To emphasize keywords, enclose them in parentheses to increase their weight. For example, '(flowers)' increases "
+        "the weight of 'flowers' by 1.1 times, '(((flowers)))' increases it by 1.331 times, and '(flowers:1.5)' increases "
+        "the weight of 'flowers' by 1.5 times. Only increase the weight for important tags. "
+        "3. The prompt should consist of three parts: prefix (quality tags + style words + effectors), subject "
+        "(main focus of the image), and scene (background, environment). The prefix affects image quality; tags like "
+        "'masterpiece,' 'best quality,' and '4k' enhance image detail, style words like 'illustration' and 'lensflare' "
+        "define the style of the image, and effectors like 'best lighting,' 'lensflare,' and 'depth of field' influence "
+        "lighting and depth. The subject is the main focus of the image, such as a character or scene. Providing a detailed "
+        "description of the subject ensures a rich and detailed image, and increasing the subject's weight enhances its clarity. "
+        "For characters, describe features like the face, hair, body, clothing, and pose. The scene describes the environment; "
+        "without a scene, the image's background appears plain, making the subject seem overly large. Some subjects inherently "
+        "include scenes (such as buildings or landscapes), and environmental words like 'flower meadow,' 'sunlight,' and 'river' "
+        "can enrich the scene. "
+        "Return only the revised prompt in English."
+    )
+
     completion = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
-            {"role": "system", "content": system_prompt},
+            {"role": "system", "content": prompt},
             {"role": "user", "content": user_input}
         ]
     )
@@ -51,11 +72,11 @@ def generate_optimized_prompt(api_key, api_base, system_prompt, user_input):
 
 def generate_fake_stream(image_url, optimized_prompt):
     chunks = [
-        "I'm generating an optimized prompt based on your input...\n",
-        f"Optimized prompt: {optimized_prompt}\n",
-        "Now generating an image based on the optimized prompt...\n",
-        "The image is being processed...\n",
-        "Almost there...\n",
+        "I'm generating an optimized prompt based on your input...",
+        f"Optimized prompt: {optimized_prompt}",
+        "Now generating an image based on the optimized prompt...",
+        "The image is being processed...",
+        "Almost there...",
         f"Image generated successfully! Here it is:\n\n![Generated Image]({image_url})"
     ]