Update app.py

app.py

@@ -10,7 +10,7 @@ generation_config = {
     "temperature": 0.7,
     "top_p": 0.95,
     "top_k": 40,
-    "max_output_tokens": 6000,
+    "max_output_tokens": 4096,
     "response_mime_type": "text/plain",
 }
 model = genai.GenerativeModel(
@@ -18,13 +18,13 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-def generate_architecture_article(images, space_characteristics):
+def generate_architecture_article(image, space_type):
     """
-    학교 공간 사진과 특징을 분석하여 건축 잡지 기사를 생성합니다.
+    학교 공간 사진과 공간 유형을 기반으로 건축 잡지 기사를 생성합니다.
 
     Args:
-        images: 학교 공간 사진 리스트 (gradio.Image 타입)
-        space_characteristics: 공간 특징 (str) - "유연하다", "연결하다", "상상하다", "특화하다" 중 하나
+        image: 학교 공간 사진 (gradio.Image 타입)
+        space_type: 공간 유형 (str) - "유연하다", "연결하다", "상상하다", "특화하다" 중 하나
 
     Returns:
         str: 생성된 건축 잡지 기사 텍스트
@@ -33,23 +33,24 @@ def generate_architecture_article(images, space_characteristics):
     full_text = ""
     yield full_text
 
-    # 이미지들을 base64로 인코딩하여 프롬프트에 추가
-    encoded_images = [gr.processing_utils.encode_base64(img) for img in images]
+    space_descriptions = {
+        "유연하다": "하나의 공간을 목적에 따라 다양한 공간으로 사용할 수 있도록 디자인된",
+        "연결하다": "공간과 공간, 사람과 사람 간 유기적인 교류와 소통을 지원하고, 정보 접근과 활용이 원활한",
+        "상상하다": "흥미와 호기심, 상상력을 자극하고, 모험적이고 놀이적 요소가 많아서 다양하고 흥미로운 경험을 제공할 수 있도록 디자인된",
+        "특화하다": "사용자가 원하는 유형의 활동(놀이, 학습, 실험, 표현 및 전시, 휴식 등)을 충분히 지원할 수 있도록 디자인된"
+    }
 
+    selected_description = space_descriptions.get(space_type, "")
 
     prompt = f"""
-    당신은 건축 전문 잡지 기자입니다. 아래 이미지들과 공간 특징을 참고하여 학교 공간에 대한 건축 잡지 기사를 작성해주세요.
-    기사에는 공간의 디자인, 기능, 분위기, 사용된 재료, 건축학적 특징, 장점과 단점 등을 포함해야 합니다.
+    당신은 건축 전문 잡지 기자입니다. 다음 이미지를 보고 학교 공간에 대한 건축 잡지 기사를 작성해주세요. 
+    이 공간은 "{selected_description}" 공간입니다.
+    기사에는 공간의 디자인, 기능, 분위기, 사용된 재료, 건축학적 특징, 장점과 단점 등을 포함해야 합니다. 
     전문적인 용어를 사용하고, 객관적인 시각을 유지하며, 비평적인 분석을 추가하여 기사를 풍부하게 만들어주세요.
 
-    **공간 특징:** {space_characteristics}
-
+    {image}
     """
 
-    for img in encoded_images:
-        prompt += f"{img}\n"
-
-
     response = model.generate_content([prompt])
 
     for chunk in response.text:
@@ -62,16 +63,12 @@ def generate_architecture_article(images, space_characteristics):
 iface = gr.Interface(
     fn=generate_architecture_article,
     inputs=[
-        gr.Gallery(label="학교 공간 사진", type="pil"),  # 여러 이미지 업로드 가능
-        gr.Radio(
-            choices=["유연하다", "연결하다", "상상하다", "특화하다"],
-            label="공간 특징",
-            info="공간의 주요 특징을 선택하세요.",
-        ),
+        gr.Image(type="pil"),
+        gr.Radio(["유연하다", "연결하다", "상상하다", "특화하다"], label="공간 분류", info="공간의 특징을 선택하세요.")
     ],
     outputs=gr.Textbox(label="건축 잡지 기사"),
     title="학교 공간 건축 기사 생성",
-    description="학교 공간 사진과 특징을 입력하여 건축 잡지 기사를 생성합니다.",
+    description="학교 공간 사진과 공간 유형을 입력하여 건축 잡지 기사를 생성합니다.",
 )
 
 # 인터페이스 실행