Update app.py

app.py

@@ -3,96 +3,92 @@ import time
 import google.generativeai as genai
 import gradio as gr
 
-# Set your Gemini API key
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
-def upload_to_gemini(path, mime_type=None):
-  """Uploads the given file to Gemini."""
-  file = genai.upload_file(path, mime_type=mime_type)
-  print(f"Uploaded file '{file.display_name}' as: {file.uri}")
-  return file
-
-def wait_for_files_active(files):
-  """Waits for the given files to be active."""
-  print("Waiting for file processing...")
-  for name in (file.name for file in files):
-    file = genai.get_file(name)
-    while file.state.name == "PROCESSING":
-      print(".", end="", flush=True)
-      time.sleep(10)
-      file = genai.get_file(name)
-    if file.state.name != "ACTIVE":
-      raise Exception(f"File {file.name} failed to process")
-  print("...all files ready")
-  print()
-
-# Create the model
+# Gemini 모델 설정
 generation_config = {
-  "temperature": 0.5,
-  "top_p": 0.95,
-  "top_k": 40,
-  "max_output_tokens": 8192,
-  "response_mime_type": "text/plain",
+    "temperature": 0.7,
+    "top_p": 0.95,
+    "top_k": 64,
+    "max_output_tokens": 5000,
+    "response_mime_type": "text/plain",
 }
-
 model = genai.GenerativeModel(
-  model_name="gemini-1.5-pro-002",
-  generation_config=generation_config,
-  system_instruction="건축가, 건축 디자이너, 학교공간혁신 전문가, 공간 및 건축관련 잡지 에디터",
+    model_name="gemini-1.5-pro-002",
+    generation_config=generation_config,
 )
 
+def generate_news_article(images, space_characteristics):
+    """
+    학교 공간 혁신 사진과 공간 특성 요소를 기반으로 기사문을 생성합니다.
+
+    Args:
+        images: 학교 공간 혁신 사진들 (gradio.Image 리스트 타입)
+        space_characteristics: 선택된 공간 특성 요소들 (리스트 타입)
+
+    Returns:
+        str: 생성된 기사문 텍스트 (한국어)
+    """
+    
+    # 초기 빈 텍스트 출력
+    full_text = ""
+    yield full_text
+
+    # 사진 설명 생성
+    image_descriptions = []
+    for i, image in enumerate(images):
+        image_descriptions.append(f"사진 {i+1}: {image}")
+
+    # 공간 특성 요소 설명 추가
+    characteristics_description = "\n**공간 특성:**\n"
+    for characteristic in space_characteristics:
+        if characteristic == "유연하다":
+            characteristics_description += "- 하나의 공간을 목적에 따라 다양한 공간으로 사용할 수 있도록 디자인\n"
+        elif characteristic == "연결하다":
+            characteristics_description += "- 공간과 공간, 사람과 사람 간 유기적인 교류와 소통을 지원하고, 정보 접근과 활용이 원활한 공간으로 디자인\n"
+        elif characteristic == "상상하다":
+            characteristics_description += "- 흥미와 호기심, 상상력을 자극하고, 모험적이고 놀이적 요소가 많아서 다양하고 흥미로운 경험을 제공할 수 있도록 디자인\n"
+        elif characteristic == "특화하다":
+            characteristics_description += "- 사용자가 원하는 유형의 활동(놀이, 학습, 실험, 표현 및 전시, 휴식 등)을 충분히 지원할 수 있도록 디자인\n"
 
-def generate_article(image_paths, space_characteristics):
-  files = []
-  for path in image_paths:
-    if path:  # Check if a file was actually uploaded
-      files.append(upload_to_gemini(path, mime_type="image/png"))  # Assuming PNG for now
+    response = model.generate_content([
+        f"""## 학교 공간 혁신, 미래 교육 환경 조성
+**사진과 선택된 공간 특성 요소들을 참고하여 학교 공간 혁신에 대한 기사문을 작성해주세요.**
 
-  if not files:
-      return "이미지를 업로드해주세요."
+{chr(10).join(image_descriptions)}
 
-  # Some files have a processing delay. Wait for them to be ready.
-  wait_for_files_active(files)
+{characteristics_description}
 
-  # Construct the prompt with space characteristics
-  prompt = "두개의 사진 자료를 참고해서 잡지 기획 기사처럼 만들어봐, 절대 교육청 홍보자료나 보고서처럼 보이면 안되고 세련된 잡지, 좋은 카피들을 좀 중간중간에 넣어서, 홈베이스 특성, 조성된 공간의 포인트 등을 잘 살려서. "
+**기사 작성 시 다음 내용을 포함해주세요:**
 
-  if space_characteristics:
-      prompt += "특히 다음과 같은 공간 특성을 반영해줘: "
-      prompt += ", ".join(space_characteristics) 
+* 혁신된 공간에 대한 구체적인 설명 (예: 스마트 교실, 협력 학습 공간, 메이커 스페이스 등)
+* 공간 특성 요소가 잘 드러난 설명
 
-  chat_session = model.start_chat(
-    history=[
-      {
-        "role": "user",
-        "parts": [
-          prompt,
-          *files  # Add all uploaded files to the prompt
-        ],
-      }
-    ]
-  )
+**전문적인 기사 형식을 갖춰 작성해주세요.**""",
+        *images 
+    ])
 
-  # Get the response directly from the started chat session
-  response = chat_session.messages[-1] 
-  return response.text
+    # 전체 텍스트를 점진적으로 생성
+    for chunk in response.text:
+        full_text += chunk
+        yield full_text
+        time.sleep(0.01)
 
 
-# Gradio 인터페이스 생성
+# Gradio 인터페이스 설정
 iface = gr.Interface(
-  fn=generate_article, 
-  inputs=[
-      gr.Image(type="filepath", label="이미지 1"), 
-      gr.Image(type="filepath", label="이미지 2"),
-      gr.Dropdown(
-          choices=["유연하다", "연결하다", "상상하다", "특화하다"], 
-          label="공간 특성", 
-          multiselect=True  # Allow multiple selections
-      )
-  ], 
-  outputs=gr.Textbox(lines=20, label="생성된 기사"),
-  title="📸 공간 기사 생성기",
-  description="사진을 업로드하면 공간에 대한 기사를 생성해줍니다.",
+    fn=generate_news_article,
+    inputs=[
+        gr.Gallery(label="학교 공간 혁신 사진들").style(grid=[2, 2]),
+        gr.Dropdown(
+            choices=["유연하다", "연결하다", "상상하다", "특화하다"],
+            label="공간 특성 요소",
+            multiselect=True
+        )
+    ],
+    outputs=gr.Textbox(label="생성된 기사문"),
+    title="학교 공간 혁신 기사 생성",
+    description="학교 공간 혁신 사진과 공간 특성 요소를 선택하여 기사문을 생성합니다.",
 )
 
 # 인터페이스 실행