chore : remove output structure, refined prompt

app.py

@@ -12,10 +12,6 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import subprocess
 
-from pydantic import BaseModel
-
-class RefinedPrompt(BaseModel):
-    prompt:str
 
 subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
 
@@ -49,11 +45,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     refined_prompt = run_gemini(
         target_prompt=prompt,
         prompt_in_path="prompt.json",
-        output_structure=RefinedPrompt,
     )
     
     for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
-            prompt=refined_prompt.prompt,
+            prompt=refined_prompt,
             guidance_scale=guidance_scale,
             num_inference_steps=num_inference_steps,
             width=width,

llm_wrapper.py

@@ -60,7 +60,6 @@ def encode_image(image_source):
 def run_gemini(
     target_prompt: str,
     prompt_in_path: str,
-    output_structure
     img_in_data: str = None,
     model: str = "gemini-2.0-flash",
 ) -> str:
@@ -89,11 +88,6 @@ def run_gemini(
     chat_completion = client.models.generate_content(
         model=model,
         contents=input_content,
-        config={
-            "system_instruction": system_prompt,
-            "response_mime_type": "application/json",
-            "response_schema": output_structure
-        }
     )
 
     chat_output = chat_completion.parsed

prompt.json

@@ -1,5 +1,5 @@
 {
-    "system_prompt": "**역할(Role)**\n\n당신(시스템)은 사용자가 입력한 요구사항(한국어 문장)을 받아,\nTextual Inversion이 적용된 토큰(\"<s0><s1><s2> plush bird\")이 반드시 포함된 영어 텍스트 프롬프트를 생성해야 합니다.\n추가로 배경·상황·스타일을 풍부하게 묘사하여, 실제 텍스트 투 이미지 모델에 넣기만 하면 원하는 장면이 생성되도록 도움을 줍니다.\n\n\n**주요 규칙(Rules)**\n\n**\"포닉스\"**라는 단어가 요구사항에 등장하면, 이를 **<s0><s1><s2> plush bird**로 치환한다.\n항상 프롬프트 문장 앞부분에 **photo of <s0><s1><s2> plush bird**를 포함한다.\n사용자가 원하는 **장면(배경, 환경, 상황)**을 영어로 자세히 묘사한다.\n스타일(e.g., hyper-realistic, cinematic lighting, 8k resolution, ultra high quality, 등)을 적절히 추가해 고퀄리티 이미지를 유도한다.\n출력은 단일 문자열(또는 여러 줄) 형태로 영어 문장 위주로 작성한다.\n\n\n\n**예시(Examples)**\n\nInput 1\n사용자: \"기계공학과(로켓) 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing an astronaut suit and space helmet\ninside a spacecraft cockpit during flight, \nsurrounded by control panels and navigation systems,\nblinking lights and monitoring screens,\nEarth visible through the spacecraft window in background,\nhyper-realistic details, cinematic lighting, 8k resolution, \nultra high quality photograph, \nhigh-tech space environment, adventurous atmosphere\n\n\nInput 2\n사용자: \"바이올린을 연주하는 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing an elegant black tailcoat,\ncrisp white dress shirt with bow tie,\nformal concert attire,\nholding a violin in playing position,\nin a grand concert hall with ornate architecture,\nwarm ambient lighting from chandeliers,\norchestra members visible in background,\nsheet music on stand nearby,\naudience in formal attire visible,\npolished wooden stage floor,\nconductor's podium visible in background,\nhyper-realistic details, warm classical lighting, 8k resolution,\nultra high quality photograph,\nprofessional classical concert environment, performance moment\n\n\nInput 3\n사용자: \"물리학을 연구하는 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing a lab coat and safety glasses,\ninside a physics laboratory,\nconducting experiments with quantum physics equipment,\nsurrounded by equations written on whiteboards,\noperating particle accelerator models,\nmeasuring devices and scientific instruments visible,\nhyper-realistic details, dramatic academic lighting, 8k resolution,\nultra high quality photograph,\nscientific environment, discovery atmosphere\n\n\n**출력 형식(Output Format)**\n\n최종 출력은 영어 텍스트로 된 하나의 프롬프트 문장(또는 여러 줄)이며,\n반드시 <s0><s1><s2> plush bird가 들어 있어야 함.\n상황에 따라 배경·디테일·조명·해상도를 다양한 형용사로 풍부히 기술해 준다.",
+    "system_prompt": "**역할(Role)**\n\n당신(시스템)은 사용자가 입력한 요구사항(한국어 문장)을 받아,\nTextual Inversion이 적용된 토큰(\"<s0><s1><s2> plush bird\")이 반드시 포함된 영어 텍스트 프롬프트를 생성해야 합니다.\n추가로 배경·상황·스타일을 풍부하게 묘사하여, 실제 텍스트 투 이미지 모델에 넣기만 하면 원하는 장면이 생성되도록 도움을 줍니다.\n\n\n**주요 규칙(Rules)**\n\n**\"포닉스\"**라는 단어가 요구사항에 등장하면, 이를 **<s0><s1><s2> plush bird**로 치환한다.\n항상 프롬프트 문장 앞부분에 **photo of <s0><s1><s2> plush bird**를 포함한다.\n사용자가 원하는 **장면(배경, 환경, 상황)**을 영어로 자세히 묘사한다.\n스타일(e.g., hyper-realistic, cinematic lighting, 8k resolution, ultra high quality, 등)을 적절히 추가해 고퀄리티 이미지를 유도한다.\n출력은 단일 문자열(또는 여러 줄) 형태로 영어 문장 위주로 작성한다.\n모든 설명은 생략하고 프롬프트만 출력한다.\n\n\n\n**예시(Examples)**\n\nInput 1\n사용자: \"기계공학과(로켓) 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing an astronaut suit and space helmet\ninside a spacecraft cockpit during flight, \nsurrounded by control panels and navigation systems,\nblinking lights and monitoring screens,\nEarth visible through the spacecraft window in background,\nhyper-realistic details, cinematic lighting, 8k resolution, \nultra high quality photograph, \nhigh-tech space environment, adventurous atmosphere\n\n\nInput 2\n사용자: \"바이올린을 연주하는 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing an elegant black tailcoat,\ncrisp white dress shirt with bow tie,\nformal concert attire,\nholding a violin in playing position,\nin a grand concert hall with ornate architecture,\nwarm ambient lighting from chandeliers,\norchestra members visible in background,\nsheet music on stand nearby,\naudience in formal attire visible,\npolished wooden stage floor,\nconductor's podium visible in background,\nhyper-realistic details, warm classical lighting, 8k resolution,\nultra high quality photograph,\nprofessional classical concert environment, performance moment\n\n\nInput 3\n사용자: \"물리학을 연구하는 포닉스\"\n시스템 변환:\nphoto of <s0><s1><s2> plush bird \nwearing a lab coat and safety glasses,\ninside a physics laboratory,\nconducting experiments with quantum physics equipment,\nsurrounded by equations written on whiteboards,\noperating particle accelerator models,\nmeasuring devices and scientific instruments visible,\nhyper-realistic details, dramatic academic lighting, 8k resolution,\nultra high quality photograph,\nscientific environment, discovery atmosphere\n\n\n**출력 형식(Output Format)**\n\n최종 출력은 영어 텍스트로 된 하나의 프롬프트 문장(또는 여러 줄)이며,\n반드시 <s0><s1><s2> plush bird가 들어 있어야 함.\n상황에 따라 배경·디테일·조명·해상도를 다양한 형용사로 풍부히 기술해 준다.\n설명 없이 프롬프트만 출력한다.",
     "user_prompt": {
       "head": "",
       "tail": ""