Update app.py

app.py

@@ -37,51 +37,53 @@ def call_api(content, system_message, max_tokens, temperature, top_p):
     )
     return response.choices[0].message['content'].strip()
 
-# 1차/2차 카테고리 선택 기능 프롬프트 생성
-def generate_category_prompt(object_input, category, background):
-    # 사물을 영어로 번역
-    system_message_object = "Translate the following object description from Korean to English. Keep it simple and concise."
-    translated_object = call_api(object_input, system_message_object, max_tokens=10)
+# 사물 및 배경 번역 함수
+def translate_object(object_input):
+    system_message = "Translate the following object description from Korean to English. Keep it simple and concise."
+    return call_api(
+        content=object_input,
+        system_message=system_message,
+        max_tokens=10,
+        temperature=0.7,
+        top_p=1.0
+    )
+
+def translate_background(background_input):
+    system_message = "Translate the following background description from Korean to English. Keep it simple and concise."
+    return call_api(
+        content=background_input,
+        system_message=system_message,
+        max_tokens=20,
+        temperature=0.7,
+        top_p=1.0
+    )
 
-    # 선택된 배경을 영어로 가져오기
+# 직접 배경 입력 탭의 프롬프트 생성 함수
+def generate_prompt(object_input, background_input):
+    translated_object = translate_object(object_input)
+    translated_background = translate_background(background_input)
+    return f"advertisement photography, {translated_object} {translated_background}"
+
+# 1차/2차 카테고리 선택 탭의 프롬프트 생성 함수
+def generate_category_prompt(object_input, category, background):
+    translated_object = translate_object(object_input)
     background_english = next(
         (item["english"] for item in categories.get(category, []) if item["korean"] == background),
         ""
     )
-
-    # 프롬프트 생성
     if translated_object and background_english:
         return f"advertisement photography, {translated_object} {background_english}."
     else:
         return "Error: Please ensure both object and background are provided correctly."
 
-# 기존 기능: 랜덤 배경 생성
-def generate_random_background(object_input):
-    # 사물을 영어로 번역
-    system_message_object = "Translate the following object description from Korean to English. Keep it simple and concise."
-    translated_object = call_api(object_input, system_message_object, max_tokens=10)
-
-    # 랜덤 배경 생성
-    system_message_background = (
-        "You are an AI that generates concise and natural background descriptions for objects in advertisement photography. "
-        "Generate only the background description and do not include 'advertisement photography' in your output. "
-        "Example 1: 'on a sunny summer beach.' "
-        "Example 2: 'on a clean wooden table near a garden.' "
-        f"Now generate a short background description for: '{translated_object}'."
-    )
-    background_prompt = call_api(translated_object, system_message_background, max_tokens=20)
-    
-    # 최종 프롬프트
-    return f"advertisement photography, {translated_object} {background_prompt}"
-
 # Gradio 인터페이스 생성
 with gr.Blocks() as demo:
     with gr.Tab("직접 배경 입력"):
-        gr.Markdown("### 한국어 이미지 프롬프트를 영어로 변환 (직접 배경 입력)")
+        gr.Markdown("### 직접 배경 입력을 통한 프롬프트 생성")
 
         with gr.Row():
             object_input = gr.Textbox(label="사물 입력", placeholder="예: 오렌지 주스")
-            background_input = gr.Textbox(label="배경 입력", placeholder="예: 깨끗한 주방 (직접 입력)")
+            background_input = gr.Textbox(label="배경 입력", placeholder="예: 깨끗한 주방")
             output = gr.Textbox(label="생성된 영어 프롬프트")
 
         generate_button = gr.Button("프롬프트 생성")
@@ -107,7 +109,6 @@ with gr.Blocks() as demo:
                 interactive=True
             )
 
-            # 1차 카테고리 선택 시 2차 배경 목록 업데이트
             def update_background_options(category):
                 if category in categories:
                     return gr.update(choices=[item["korean"] for item in categories[category]], value=None)
@@ -129,18 +130,4 @@ with gr.Blocks() as demo:
             outputs=category_prompt_output
         )
 
-    with gr.Tab("랜덤 배경 생성"):
-        gr.Markdown("### 한국어 이미지 프롬프트를 영어로 변환 (랜덤 배경 생성)")
-
-        with gr.Row():
-            object_input = gr.Textbox(label="사물 입력", placeholder="예: 오렌지 주스")
-            random_background_output = gr.Textbox(label="자동 생성된 영어 프롬프트")
-
-        random_generate_button = gr.Button("랜덤 배경 생성")
-        random_generate_button.click(
-            fn=generate_random_background,
-            inputs=object_input,
-            outputs=random_background_output
-        )
-
 demo.launch()