Update app.py

app.py

@@ -2,8 +2,7 @@ import gradio as gr
 import google.generativeai as genai
 import os
 
-# --- 1. 챗봇의 역할(시스템 프롬프트)을 여기에 정의합니다. ---
-# 예시: IT 및 코딩 질문에 전문적으로 답변하는 AI 어시스턴트
+# 시스템 프롬프트 정의
 SYSTEM_PROMPT = "발표 대본을 만들어주는 챗봇입니다. 발표할 내용을 입력하면 '왜냐하면 ~이기 때문입니다'로 대본을 만들어줍니다"
 
 # API 키 설정
@@ -12,18 +11,12 @@ if api_key:
     genai.configure(api_key=api_key)
 
 def chat(message, history):
-    # --- 수정 1: 들여쓰기 수정 ---
-    # if 문 아래의 모든 코드는 4칸 들여쓰기가 필요합니다
     if not api_key:
-        return "❌ API 키가 설정되지 않았습니다. Hugging Face Spaces의 Settings에서 GEMINI_API_KEY를 Secret으로 추가하세요."
+        return history + [[message, "❌ API 키가 설정되지 않았습니다."]]
     
-    # --- 수정 2: try 블록 들여쓰기 수정 ---
-    # try 블록이 if 문과 동일한 레벨에 있도록 수정
     try:
-        # --- 수정 3: 모델명 변경 ---
-        # 'gemini-2.0-flash' → 'gemini-1.5-flash'로 변경 (현재 사용 가능한 모델)
         model = genai.GenerativeModel(
-            'gemini-1.5-flash',  # 수정: gemini-2.0-flash는 아직 사용 불가
+            'gemini-1.5-flash',
             system_instruction=SYSTEM_PROMPT
         )
         
@@ -35,28 +28,59 @@ def chat(message, history):
             if assistant:
                 chat_history.append({"role": "model", "parts": [assistant]})
         
-        # 채팅 세션 시작
+        # 채팅 세션 시작 및 응답 생성
         chat_session = model.start_chat(history=chat_history)
-        
-        # 응답 생성
         response = chat_session.send_message(message)
-        return response.text
+        
+        # 히스토리에 추가
+        history.append([message, response.text])
+        return history
         
     except Exception as e:
-        return f"❌ 오류 발생: {str(e)}"
+        history.append([message, f"❌ 오류 발생: {str(e)}"])
+        return history
 
-# --- 수정 4: examples 제거 및 type 추가 ---
-# examples 파라미터는 Gradio 버그를 일으킬 수 있으므로 제거
-# type="messages"를 추가하여 경고 메시지 해결
-demo = gr.ChatInterface(
-    fn=chat,
-    title="🤖 논리적으로 말하는 발표도우미 Gemini 챗봇",
-    description="논리적으로 발표하는 방법을 도와줍니다.",
-    # examples=["학급회의때 말하는 방법", "국어시간에 주장하는 말 하는 방법", "사회현상에 대한 의견"],  # 제거
-    type="messages"  # 추가: deprecation 경고 해결
-)
+# Gradio Blocks 인터페이스
+with gr.Blocks() as demo:
+    gr.Markdown("""
+    # 🤖 논리적으로 말하는 발표도우미 Gemini 챗봇
+    논리적으로 발표하는 방법을 도와줍니다.
+    """)
+    
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(
+        placeholder="발표할 내용을 입력하세요...",
+        label="메시지 입력"
+    )
+    
+    with gr.Row():
+        submit = gr.Button("전송", variant="primary")
+        clear = gr.Button("대화 초기화")
+    
+    # 예제 섹션
+    gr.Markdown("### 💡 예제 질문")
+    with gr.Row():
+        ex1 = gr.Button("학급회의때 말하는 방법", size="sm")
+        ex2 = gr.Button("국어시간에 주장하는 말 하는 방법", size="sm")
+        ex3 = gr.Button("사회현상에 대한 의견", size="sm")
+    
+    # 이벤트 핸들러
+    def respond(message, chat_history):
+        if message.strip():
+            return chat(message, chat_history), ""
+        return chat_history, ""
+    
+    # 메시지 전송
+    msg.submit(respond, [msg, chatbot], [chatbot, msg])
+    submit.click(respond, [msg, chatbot], [chatbot, msg])
+    
+    # 대화 초기화
+    clear.click(lambda: [], None, chatbot)
+    
+    # 예제 버튼 클릭
+    ex1.click(lambda h: chat("학급회의때 말하는 방법", h), [chatbot], [chatbot])
+    ex2.click(lambda h: chat("국어시간에 주장하는 말 하는 방법", h), [chatbot], [chatbot])
+    ex3.click(lambda h: chat("사회현상에 대한 의견", h), [chatbot], [chatbot])
 
-# --- 수정 5: __name__ 수정 ---
-# **name** → __name__ (언더스코어 두 개씩)
 if __name__ == "__main__":
     demo.launch()