Update src/streamlit_app.py
Browse files- src/streamlit_app.py +3 -5
src/streamlit_app.py
CHANGED
|
@@ -81,10 +81,8 @@ def get_current_system_prompt():
|
|
| 81 |
st.session_state["assigned_prompt_type"] = random.choice(["dense", "bullet"])
|
| 82 |
|
| 83 |
if st.session_state["assigned_prompt_type"] == "dense":
|
| 84 |
-
print("111")
|
| 85 |
return SYSTEM_PROMPT_DENSE
|
| 86 |
else:
|
| 87 |
-
print("222")
|
| 88 |
return SYSTEM_PROMPT_BULLET
|
| 89 |
|
| 90 |
|
|
@@ -161,9 +159,9 @@ Write the summary following the rules in the system prompt.
|
|
| 161 |
# ------------------------------------------------
|
| 162 |
# Simple chat UI (original style)
|
| 163 |
# ------------------------------------------------
|
| 164 |
-
st.set_page_config(page_title="Simple LLM", page_icon="
|
| 165 |
|
| 166 |
-
st.title("
|
| 167 |
|
| 168 |
|
| 169 |
# Initialize chat history
|
|
@@ -194,7 +192,7 @@ for msg in st.session_state["messages"]:
|
|
| 194 |
st.markdown(msg["content"])
|
| 195 |
|
| 196 |
# Chat input (simple, original format)
|
| 197 |
-
user_input = st.chat_input("
|
| 198 |
|
| 199 |
if user_input:
|
| 200 |
# Add user message
|
|
|
|
| 81 |
st.session_state["assigned_prompt_type"] = random.choice(["dense", "bullet"])
|
| 82 |
|
| 83 |
if st.session_state["assigned_prompt_type"] == "dense":
|
|
|
|
| 84 |
return SYSTEM_PROMPT_DENSE
|
| 85 |
else:
|
|
|
|
| 86 |
return SYSTEM_PROMPT_BULLET
|
| 87 |
|
| 88 |
|
|
|
|
| 159 |
# ------------------------------------------------
|
| 160 |
# Simple chat UI (original style)
|
| 161 |
# ------------------------------------------------
|
| 162 |
+
st.set_page_config(page_title="Simple LLM", page_icon="")
|
| 163 |
|
| 164 |
+
st.title("Simple LLM")
|
| 165 |
|
| 166 |
|
| 167 |
# Initialize chat history
|
|
|
|
| 192 |
st.markdown(msg["content"])
|
| 193 |
|
| 194 |
# Chat input (simple, original format)
|
| 195 |
+
user_input = st.chat_input("Ask something...")
|
| 196 |
|
| 197 |
if user_input:
|
| 198 |
# Add user message
|