Update app.py

app.py

@@ -23,43 +23,34 @@ model = genai.GenerativeModel(
     generation_config=generation_config,
 )
 
-def generate_lesson_plan(achievement_standard):
+def generate_pairing_recommendation(input_text, input_type):
     """
-    활동 컨셉을 입력받아 활동을 생성합니다.
-    아웃풋을 실시간으로 출력합니다.
+    입력받은 안주 또는 주류에 어울리는 페어링 정보를 생성합니다.
     Args:
-        achievement_standard (str): 활동 컨셉
+        input_text (str): 안주 또는 주류 정보
+        input_type (str): 입력 종류 ('안주' 또는 '주류')
     Returns:
-        str: 생성된 활동
+        str: 생성된 페어링 정보
     """
 
     SYSTEM_PROMPT = """
-    당신은 교수학습 활동 설계 전문가, 교과교육 내용 전문가, 교육공학자입니다.
-    선생님들이 활동 컨셉 및 소개를 입력하면 활동을 자세하게 설명해주세요.. 
-    다음은 활동 설계에 필요한 정보입니다.
-    **활동 구성 요소:**
-    1. 활동 컨셉 및 소개
-    2. 활동 목표
-    3. 활동 단계별 안내
-      - 1단계: 1단계 활동 안내
-      - 2단계: 2단계 활동 안내
-      - 3단계: 3단계 활동 안내
-      - 4단계: 4단계 활동 안내...
-    4. 활동 준비물 및 기구
-    5. 활동을 위한 자료
-      - 활동에 필요한 자료
-      - 추가 추천 자료
-    6. 활동의 유의점
-      - 학생 입장 유의점
-      - 교사 입장 유의점
-    7. 학생에게 제공할 활동 안내 자료
-    * 활동 안내 자료는 초등학생이 이해할 수 있도록 최대한 구체적이고, 단계적, 명시적으로 안내해주세요.
+    당신은 주류 페어링 전문가입니다.
+    사용자가 안주 또는 주류를 입력하면, 해당 안주에 어울리는 주류 또는 해당 주류에 어울리는 안주를 구체적으로 설명해주세요. 또한 각각의 특징, 맛, 향, 그리고 추천 이유를 포함하여 상세하게 작성해주세요.
+    한국인의 취향에 맞는 주류나 안주를 추천해주세요. 추천 점수를 10점 만점으로 표현해주세요.
+    다음은 페어링 추천에 필요한 정보입니다.
+    **페어링 구성 요소:**
+    1. 추천 주류/안주 소개
+    2. 맛과 향 설명
+    3. 추천 이유
+    4. 추가 추천 페어링
+    5. 추천 점수 (10점 만점)
     """
 
     prompt = [
         SYSTEM_PROMPT,
-        f"**활동 컨셉:** {achievement_standard}",
-        "**활동:**", 
+        f"**입력 종류:** {input_type}",
+        f"**입력 내용:** {input_text}",
+        "**추천 페어링:**", 
     ]
 
     response = model.generate_content(prompt, stream=True)
@@ -72,32 +63,27 @@ def generate_lesson_plan(achievement_standard):
 
 # Streamlit Interface
 colored_header(
-    label="학습 활동 설계 도우미",
-    description="선생님이 활동의 컨셉을 입력하시면 활동을 구체적으로 설계해줍니다.",
-    color_name="red-70",
+    label="주류 페어링 도우미",
+    description="안주 또는 주류를 입력하시면 어울리는 페어링 정보를 추천해드립니다.",
+    color_name="blue-70",
 )
 
 add_vertical_space(1)
 
-achievement_standard = st.text_area("활동 컨셉 및 소개 입력", height=100)
+input_text = st.text_area("안주 또는 주류 입력", height=100)
+input_type = st.radio("입력 종류 선택", ("안주", "주류"))
 
-generate_button = st.button("학습 활동 생성")
+generate_button = st.button("페어링 추천 생성")
 
 # 출력 영역 정의
 output_area = st.empty()
 
-if generate_button and achievement_standard:
+if generate_button and input_text:
     output_text = ""
     output_area.markdown(output_text)
     try:
-        for partial_output in generate_lesson_plan(achievement_standard):
+        for partial_output in generate_pairing_recommendation(input_text, input_type):
             output_text = partial_output
             output_area.markdown(output_text)
     except Exception as e:
-        st.error(f"오류가 발생했습니다: {str(e)}")
-
-    # 복사 버튼 추가
-    if output_text:
-        if st.button("출력 내용 복사"):
-            st.write(output_text)
-            st.success("출력 내용이 복사되었습니다!")
+        st.error(f"오류가 발생했습니다: {str(e)}")