Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
+import time
 import google.generativeai as genai
 import gradio as gr
-from PIL import Image
 
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
@@ -9,80 +9,70 @@ genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 generation_config = {
     "temperature": 0.7,
     "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 5000,
+    "top_k": 40,
+    "max_output_tokens": 6000,
     "response_mime_type": "text/plain",
 }
 model = genai.GenerativeModel(
-    model_name="gemini-1.5-pro-002",
+    model_name="gemini-1.5-flash-exp-0827",
     generation_config=generation_config,
 )
 
-def generate_news_article(images, space_characteristic, key_features):
+def generate_architecture_article(images, space_characteristics):
     """
-    학교 공간 혁신 사진, 공간 특성, 주요 특징을 기반으로 기사문을 생성합니다.
+    학교 공간 사진과 특징을 분석하여 건축 잡지 기사를 생성합니다.
+
+    Args:
+        images: 학교 공간 사진 리스트 (gradio.Image 타입)
+        space_characteristics: 공간 특징 (str) - "유연하다", "연결하다", "상상하다", "특화하다" 중 하나
+
+    Returns:
+        str: 생성된 건축 잡지 기사 텍스트
+    """
+
+    full_text = ""
+    yield full_text
+
+    # 이미지들을 base64로 인코딩하여 프롬프트에 추가
+    encoded_images = [gr.processing_utils.encode_base64(img) for img in images]
+
+
+    prompt = f"""
+    당신은 건축 전문 잡지 기자입니다. 아래 이미지들과 공간 특징을 참고하여 학교 공간에 대한 건축 잡지 기사를 작성해주세요.
+    기사에는 공간의 디자인, 기능, 분위기, 사용된 재료, 건축학적 특징, 장점과 단점 등을 포함해야 합니다.
+    전문적인 용어를 사용하고, 객관적인 시각을 유지하며, 비평적인 분석을 추가하여 기사를 풍부하게 만들어주세요.
+
+    **공간 특징:** {space_characteristics}
+
     """
 
-    formatted_descriptions = ""
-    if images:
-        for i, image_path in enumerate(images):
-            try:
-                # 이미지 열기
-                image = Image.open(image_path)
-                description = genai.generate_image_caption(image)
-                formatted_descriptions += f"사진 {i+1} 설명: {description.result}\n"
-            except Exception as e:
-                print(f"Error generating caption: {e}")
-                formatted_descriptions += f"사진 {i+1} 설명: (설명 생성 오류)\n"
-    else:
-        formatted_descriptions = "사진 설명 없음\n"
-
-
-    characteristic_description = ""
-    if space_characteristic:
-        characteristic_description = f"\n**공간 특성: {space_characteristic}**\n"
-        if space_characteristic == "유연하다":
-            characteristic_description += "- 하나의 공간을 목적에 따라 다양한 공간으로 사용할 수 있도록 디자인\n"
-        elif space_characteristic == "연결하다":
-            characteristic_description += "- 공간과 공간, 사람과 사람 간 유기적인 교류와 소통을 지원하고, 정보 접근과 활용이 원활한 공간으로 디자인\n"
-        elif space_characteristic == "상상하다":
-            characteristic_description += "- 흥미와 호기심, 상상력을 자극하고, 모험적이고 놀이적 요소가 많아서 다양하고 흥미로운 경험을 제공할 수 있도록 디자인\n"
-        elif space_characteristic == "특화하다":
-            characteristic_description += "- 사용자가 원하는 유형의 활동(놀이, 학습, 실험, 표현 및 전시, 휴식 등)을 충분히 지원할 수 있도록 디자인\n"
-
-
-    prompt = f"""## 학교 공간 혁신 기사 작성
-{formatted_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)
-        full_text = response.result
-    except Exception as e:
-        print(f"Error generating text: {e}")
-        full_text = "기사 생성 오류"
-
-    return full_text
+    for img in encoded_images:
+        prompt += f"{img}\n"
+
+
+    response = model.generate_content([prompt])
+
+    for chunk in response.text:
+        full_text += chunk
+        yield full_text
+        time.sleep(0.01)
 
 
+# Gradio 인터페이스 설정
 iface = gr.Interface(
-    fn=generate_news_article,
+    fn=generate_architecture_article,
     inputs=[
-        gr.Files(label="사진 업로드 (여러 장 가능)", type="filepath"),
-        gr.Radio(["유연하다", "연결하다", "상상하다", "특화하다"], label="공간 특성"),
-        gr.Textbox(label="주요 특징", lines=3),
+        gr.Gallery(label="학교 공간 사진", type="pil"),  # 여러 이미지 업로드 가능
+        gr.Radio(
+            choices=["유연하다", "연결하다", "상상하다", "특화하다"],
+            label="공간 특징",
+            info="공간의 주요 특징을 선택하세요.",
+        ),
     ],
-    outputs=gr.Textbox(label="생성된 기사", lines=20),
-    title="학교 공간 혁신 기사 생성",
-    description="사진, 공간 특성, 주요 특징을 입력하여 학교 공간 혁신 기사를 생성합니다.",
+    outputs=gr.Textbox(label="건축 잡지 기사"),
+    title="학교 공간 건축 기사 생성",
+    description="학교 공간 사진과 특징을 입력하여 건축 잡지 기사를 생성합니다.",
 )
 
+# 인터페이스 실행
 iface.launch()