Create app.py

app.py file for solar chat with your very own system prompt!

app.py

@@ -0,0 +1,101 @@
+import streamlit as st
+import pandas as pd
+from io import StringIO
+from datetime import datetime
+import requests
+import sseclient  # for streaming response
+
+st.set_page_config(page_title="Solar Chat", page_icon="🌞")
+st.title("🌞 Chat with your own System Prompt")
+
+# --- User Input ---
+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
+
+# --- Solar Pro API call function ---
+def solar_pro_chat(messages, api_key):
+    url = "https://api.upstage.ai/v1"
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Accept": "text/event-stream",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "model": "solar-pro", 
+        "messages": messages,
+        "stream": True
+    }
+
+    # 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
+
+# --- 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})
+
+# --- 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)
+
+st.markdown("### 🗂️ Chat History")
+for msg in st.session_state.messages:
+    if msg["role"] != "system":
+        render_message(msg["role"], msg["content"])
+
+# --- CSV Download Function ---
+def generate_csv():
+    rows = []
+    rows.append({"role": "system", "content": st.session_state.system_prompt})
+    for msg in st.session_state.messages:
+        if msg["role"] != "system":
+            rows.append(msg)
+    df = pd.DataFrame(rows)
+    output = StringIO()
+    df.to_csv(output, index=False)
+    return output.getvalue()
+
+now = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+filename = f"solar_chat_{now}.csv"
+
+st.download_button(
+    label="⬇️ Download Chat History as CSV",
+    data=generate_csv(),
+    file_name=filename,
+    mime="text/csv",
+)