mod text2zip

backend/app/app/api/api_v1/endpoints/text2zip.py

@@ -48,15 +48,19 @@ LINE_TOKEN = 'o8dqdVL2k8aiWO4jy3pawZamBu53bbjoSh2u0GJ7F0j'
 router = APIRouter()
 
 def gen_prompt(content:str):
-    client = OpenAI(api_key='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')
+    client = OpenAI(base_url="http://192.168.192.84:8080/v1", api_key='choozmo9')
 
     completion = client.chat.completions.create(
-        model="gpt-4o-mini",
+        model="gemma",
         messages=[
-            {"role": "assistant", "content": "You are a helpful image genaration prompt engineer. \
-             You will convert the following inputs into English prompts for image generation AI and respond accordingly. \
-             Do not start with 'Create an image. \
-             Keep it within 50 words"},
+            {"role": "system", "content": "You are a helpful image generation prompt engineer. "
+                                          "You will convert the following inputs into English prompt for image generation AI and respond accordingly. "
+                                          "Do not start with 'Create an image'. "
+                                          "Don't ask me more information. "
+                                          "Directly answer to me. "
+                                          "If the input is too long, summarize it. "
+                                          "If the input is too short, expand it with reasonable imagination."
+                                          "Keep the prompt within 70 tokens. "},
             {
                 "role": "user",
                 "content": content
@@ -65,8 +69,26 @@ def gen_prompt(content:str):
     )
     return completion.choices[0].message.content
 
+def gen_sd_image(prompt):
+    api = webuiapi.WebUIApi(host='192.168.192.84', port=8081)
+    api.util_set_model('sd3_medium')
+    result = api.txt2img(prompt=prompt,
+                    negative_prompt="((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), wierd colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), ((ugly)), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy",
+                    seed=-1,
+                    cfg_scale=4,
+                    sampler_name='Euler',
+                    scheduler='Automatic',
+                    steps=40,
+                    width=1280,
+                    height=720,
+                    )
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as t:
+        result.image.save(t, "jpeg")
+    return t.name
+
+
 def gen_flux_image(prompt):
-    client = Client("http://192.168.192.83:7860/")
+    client = Client("http://192.168.192.84:8082/")
     result = client.predict(
         model_id="models/FLUX.1-schnell",
         prompt=prompt,
@@ -82,23 +104,6 @@ def gen_flux_image(prompt):
         PIL.Image.open(result[0]).convert("RGB").save(t.name,"jpeg")
         return t.name
 
-def gen_sd_image(prompt):
-    api = webuiapi.WebUIApi(host='192.168.192.38', port=7860)
-    api.util_set_model('sd3_medium')
-    result = api.txt2img(prompt=prompt,
-                    negative_prompt="",
-                    seed=-1,
-                    cfg_scale=4,
-                    sampler_name='Euler',
-                    scheduler='Automatic',
-                    steps=40,
-                    width=1280,
-                    height=720,
-                    )
-    with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as t:
-        result.image.save(t, "jpeg")
-    return t.name
-
 punctuation = r"[.,!?;:。 、！？，；：]"