Update app.py

app.py

@@ -1,9 +1,11 @@
 import os
 import time
 import google.generativeai as genai
-import gradio as gr
+import streamlit as st
 import re
+from streamlit_chat import message
 
+# Google Generative AI API 설정
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 # 모델 설정
@@ -24,7 +26,6 @@ safety_settings = [
 # 전역 변수로 chat_session 초기화
 chat_session = None
 
-
 def start_new_chat_session():
     global chat_session
     chat_session = (
@@ -39,18 +40,29 @@ def start_new_chat_session():
         .start_chat(history=[])
     )
 
-
 # 초기 세션 시작
 start_new_chat_session()
 
+# Streamlit 앱 설정
+st.set_page_config(page_title="AI 과학자", layout="wide")
+st.markdown("<div style='font-size: 30px; font-weight: bold;'>AI 과학자</div>", unsafe_allow_html=True)
+st.markdown("과학에 대해 궁금한 점을 AI 과학자에게 질문하세요, 자유탐구에 대한 조언을 구하세요.")
 
-def respond(user_input, history):
-    global chat_session
-    history.append((user_input, ""))
-    yield "", history  # 초기 빈 텍스트 출력
+# 채팅 히스토리 관리
+if "history" not in st.session_state:
+    st.session_state.history = []
 
+# 채팅 세션 초기화
+if "chat_session" not in st.session_state:
+    st.session_state.chat_session = chat_session
+
+# 사용자 입력 받기
+user_input = st.text_input("입력", key="user_input")
+
+def respond(user_input):
+    st.session_state.history.append({"user": user_input, "ai": ""})
     try:
-        response = chat_session.send_message(user_input, stream=True)
+        response = st.session_state.chat_session.send_message(user_input, stream=True)
         full_text = ""
         
         # 스트리밍 응답 처리
@@ -66,32 +78,25 @@ def respond(user_input, history):
                 else:
                     full_text += sentence
                 
-                history[-1] = (user_input, full_text)
-                yield "", history
+                # 히스토리 최신 항목 업데이트
+                st.session_state.history[-1]["ai"] = full_text
                 time.sleep(0.03)  # 출력 속도 조절
 
     except Exception as e:
-        yield f"에러 발생: {str(e)}" 
-
+        st.session_state.history[-1]["ai"] = f"에러 발생: {str(e)}"
 
-def clear_chat():
-    start_new_chat_session()
-    return []
-
-
-with gr.Blocks() as demo:
-    gr.Markdown(
-        "<div style='font-size: 30px; font-weight: bold;'>AI 과학자</div>"
-    )
-    gr.Markdown(
-        "과학에 대해 궁금한 점을 AI 과학자에게 질문하세요, 자유탐구에 대한 조언을 구하세요."
-    )
-    chatbot = gr.Chatbot(label="채팅창")
-    msg = gr.Textbox(label="입력")
-    clear = gr.Button("초기화")
+# 응답 생성 및 출력
+if user_input:
+    respond(user_input)
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    # chatbot 상태 초기화 및 새로운 세션 시작
-    clear.click(clear_chat, outputs=chatbot, queue=False)
+# 채팅 메시지 출력
+for chat in st.session_state.history:
+    if "user" in chat:
+        message(chat["user"], is_user=True, key=f"user_{chat['user']}")
+    if "ai" in chat:
+        message(chat["ai"], is_user=False, key=f"ai_{chat['ai']}")
 
-demo.launch(share=True)
+# 초기화 버튼 클릭 시
+if st.button("초기화"):
+    start_new_chat_session()
+    st.session_state.history = []