Update app.py

app.py

@@ -18,21 +18,20 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-def generate_news_article(images, space_characteristic):
+def generate_news_article(images, space_characteristic, key_features):
     """
-    학교 공간 혁신 사진과 공간 특성 요소를 기반으로 기사문을 생성합니다.
+    학교 공간 혁신 사진, 공간 특성, 주요 특징을 기반으로 기사문을 생성합니다.
     """
-    
+
     image_descriptions = []
     for i, image in enumerate(images):
         try:
-             image_description = genai.describe_image(image)
-             image_descriptions.append(f"사진 {i+1} 설명: {image_description.description}")
+            image_description = genai.describe_image(image)
+            image_descriptions.append(f"사진 {i+1} 설명: {image_description.description}")
         except Exception as e:
             print(f"Error describing image {i+1}: {e}")
             image_descriptions.append(f"사진 {i+1} 설명: 오류 발생")
 
-
     characteristic_description = ""
     if space_characteristic:
         characteristic_description = f"\n**공간 특성: {space_characteristic}**\n"
@@ -45,7 +44,25 @@ def generate_news_article(images, space_characteristic):
         elif space_characteristic == "특화하다":
             characteristic_description += "- 사용자가 원하는 유형의 활동(놀이, 학습, 실험, 표현 및 전시, 휴식 등)을 충분히 지원할 수 있도록 디자인\n"
 
-    prompt = f"## 학교 공간 혁신 기사 작성\n\n{''.join(image_descriptions)}\n{characteristic_description}\n위 내용을 바탕으로 학교 공간 혁신에 대한 기사를 작성해주세요."
+
+    prompt = f"""## 학교 공간 혁신 기사 작성
+
+{''.join(image_descriptions)}
+
+{characteristic_description}
+
+**주요 특징:** {key_features}
+
+위 내용을 바탕으로 **{space_characteristic}** 공간 특성을 반영하여 다음과 같은 서식을 갖춘 학교 공간 혁신에 대한 기사를 작성해주세요.
+
+**제목:** ({space_characteristic} 특성을 반영한 제목)
+
+**소제목 1:** {space_characteristic} 공간의 특징을 잘 드러내는 소제목
+({space_characteristic} 특성을 반영한 내용)
+
+**소제목 2:** {space_characteristic} 공간의 다른 특징을 잘 드러내는 소제목
+({space_characteristic} 특성을 반영한 내용)
+"""
 
     try:
         response = model.generate_text(prompt=prompt)
@@ -53,19 +70,20 @@ def generate_news_article(images, space_characteristic):
     except Exception as e:
         print(f"Error generating text: {e}")
         full_text = "기사 생성 오류"
-        
+
     return full_text
 
 
 with gr.Blocks() as demo:
     image_inputs = gr.Gallery(label="학교 공간 사진 업로드", type="pil")
     space_characteristic = gr.Radio(["유연하다", "연결하다", "상상하다", "특화하다"], label="공간 특성")
+    key_features = gr.Textbox(label="주요 특징", lines=3)
     generate_btn = gr.Button("기사 생성")
     news_article = gr.Textbox(label="생성된 기사", lines=20)
 
     generate_btn.click(
         fn=generate_news_article,
-        inputs=[image_inputs, space_characteristic],
+        inputs=[image_inputs, space_characteristic, key_features],
         outputs=news_article,
     )