Update app.py

system comparison function added

app.py

@@ -4,22 +4,29 @@ import json
 from io import StringIO
 from datetime import datetime
 import requests
-import sseclient  # for streaming response
+import sseclient
 
-st.set_page_config(page_title="Solar Chat", page_icon="🌞")
-st.title("🌞 Chat with your own System Prompt")
+st.set_page_config(page_title="Solar Prompt Comparator", page_icon="🌞")
+st.title("🌞 System Prompt Comparison Chat")
 
-# --- User Input ---
+# --- Input UI ---
 api_key = st.text_input("🔑 Upstage API Key (starts with 'up_')", type="password")
-system_prompt = st.text_area("🧠 System Prompt", "You are a helpful assistant.", height=100)
 user_input = st.text_input("💬 Your Message", "")
 
-# --- Session state Reset ---
-if "messages" not in st.session_state or st.session_state.system_prompt != system_prompt:
-    st.session_state.messages = [{"role": "system", "content": system_prompt}]
-    st.session_state.system_prompt = system_prompt
+st.markdown("### 🧠 Custom System Prompt")
+custom_prompt = st.text_area("사용자 정의 시스템 프롬프트", "You are a helpful assistant.", height=100)
 
-# --- Solar Pro API call function ---
+# --- Default prompt 설정 ---
+default_prompt = "You are a helpful assistant."
+
+# --- 세션 초기화 ---
+if "default_messages" not in st.session_state:
+    st.session_state.default_messages = [{"role": "system", "content": default_prompt}]
+if "custom_messages" not in st.session_state or st.session_state.custom_prompt != custom_prompt:
+    st.session_state.custom_messages = [{"role": "system", "content": custom_prompt}]
+    st.session_state.custom_prompt = custom_prompt
+
+# --- Solar Pro API 호출 함수 ---
 def solar_pro_chat(messages, api_key):
     url = "https://api.upstage.ai/v1/chat/completions"
     headers = {
@@ -30,84 +37,59 @@ def solar_pro_chat(messages, api_key):
     payload = {
         "model": "solar-pro", 
         "messages": messages,
-        "stream": True  # SSE 방식 활성화
+        "stream": True
     }
 
     try:
         response = requests.post(url, headers=headers, json=payload, stream=True)
-        response.raise_for_status()  # HTTP 에러 발생 시 예외 발생
-
+        response.raise_for_status()
         client = sseclient.SSEClient(response)
-        full_reply = ""
-
         for event in client.events():
             if event.data == "[DONE]":
                 break
             try:
                 content = json.loads(event.data)["choices"][0]["delta"].get("content", "")
-                full_reply += content
                 yield content
             except Exception as e:
-                st.error("⚠️ 응답 처리 중 오류가 발생했습니다. 로그를 확인하세요.")
+                st.error("⚠️ 응답 처리 중 오류가 발생했습니다.")
                 print(f"[SSE 파싱 오류] {e}")
                 continue
-
     except requests.exceptions.RequestException as e:
-        st.error("❌ API 호출 중 문제가 발생했습니다. API 키를 확인하거나 네트워크 상태를 점검해주세요.")
+        st.error("❌ API 호출 실패: API 키 또는 네트워크 상태 확인")
         print(f"[API 요청 오류] {e}")
         return
 
-    # SSE streaming
-    response = requests.post(url, headers=headers, json=payload, stream=True)
-    client = sseclient.SSEClient(response)
-    full_reply = ""
-    for event in client.events():
-        if event.data == "[DONE]":
-            break
-        try:
-            content = eval(event.data)["choices"][0]["delta"].get("content", "")
-            full_reply += content
-            yield content
-        except Exception:
-            continue
-    return full_reply
+# --- 비교용 응답 처리 ---
+if st.button("🚀 Compare Responses") and api_key and user_input:
+    st.session_state.default_messages.append({"role": "user", "content": user_input})
+    st.session_state.custom_messages.append({"role": "user", "content": user_input})
 
-# --- Send Messages ---
-if st.button("Send") and api_key and user_input:
-    st.session_state.messages.append({"role": "user", "content": user_input})
-    with st.spinner("Generating response..."):
-        response_text = ""
-        response_area = st.empty()
-        for chunk in solar_pro_chat(st.session_state.messages, api_key):
-            response_text += chunk
-            response_area.markdown(f"**🤖 Bot:** {response_text}")
-        st.session_state.messages.append({"role": "assistant", "content": response_text})
+    col1, col2 = st.columns(2)
 
-# --- Chat history in bubbles ---
-def render_message(role, content):
-    if role == "user":
-        st.markdown(f"""
-        <div style='background-color:#DCF8C6; padding:10px 15px; border-radius:10px; margin:5px 0; text-align:right;'>
-            <b>You:</b> {content}
-        </div>
-        """, unsafe_allow_html=True)
-    elif role == "assistant":
-        st.markdown(f"""
-        <div style='background-color:#F1F0F0; padding:10px 15px; border-radius:10px; margin:5px 0; text-align:left;'>
-            <b>Bot:</b> {content}
-        </div>
-        """, unsafe_allow_html=True)
+    with col1:
+        st.subheader("🔹 Default Prompt")
+        default_response = ""
+        default_area = st.empty()
+        with st.spinner("Default 응답 생성 중..."):
+            for chunk in solar_pro_chat(st.session_state.default_messages, api_key):
+                default_response += chunk
+                default_area.markdown(f"**🤖 Bot:** {default_response}")
+        st.session_state.default_messages.append({"role": "assistant", "content": default_response})
 
-st.markdown("### 🗂️ Chat History")
-for msg in st.session_state.messages:
-    if msg["role"] != "system":
-        render_message(msg["role"], msg["content"])
+    with col2:
+        st.subheader("🔸 Custom Prompt")
+        custom_response = ""
+        custom_area = st.empty()
+        with st.spinner("Custom 응답 생성 중..."):
+            for chunk in solar_pro_chat(st.session_state.custom_messages, api_key):
+                custom_response += chunk
+                custom_area.markdown(f"**🤖 Bot:** {custom_response}")
+        st.session_state.custom_messages.append({"role": "assistant", "content": custom_response})
 
-# --- CSV Download Function ---
-def generate_csv():
-    rows = []
-    rows.append({"role": "system", "content": st.session_state.system_prompt})
-    for msg in st.session_state.messages:
+# --- 채팅 히스토리 다운로드 ---
+def generate_csv(messages, prompt_label):
+    rows = [{"role": "system", "content": prompt_label}]
+    for msg in messages:
         if msg["role"] != "system":
             rows.append(msg)
     df = pd.DataFrame(rows)
@@ -116,11 +98,20 @@ def generate_csv():
     return output.getvalue()
 
 now = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
-filename = f"solar_chat_{now}.csv"
+st.markdown("### ⬇️ Download Chat Histories")
+col1, col2 = st.columns(2)
+with col1:
+    st.download_button(
+        label="Download Default Chat",
+        data=generate_csv(st.session_state.default_messages, default_prompt),
+        file_name=f"default_chat_{now}.csv",
+        mime="text/csv",
+    )
+with col2:
+    st.download_button(
+        label="Download Custom Chat",
+        data=generate_csv(st.session_state.custom_messages, custom_prompt),
+        file_name=f"custom_chat_{now}.csv",
+        mime="text/csv",
+    )
 
-st.download_button(
-    label="⬇️ Download Chat History as CSV",
-    data=generate_csv(),
-    file_name=filename,
-    mime="text/csv",
-)