neuralleap commited on
Commit
c416d39
·
verified ·
1 Parent(s): facfb6b
Files changed (1) hide show
  1. app.py +20 -12
app.py CHANGED
@@ -4,7 +4,6 @@ from dotenv import load_dotenv
4
  import os
5
  import uuid
6
  from datetime import datetime
7
- import time
8
 
9
  # Load environment variables
10
  load_dotenv()
@@ -132,14 +131,14 @@ def create_new_chat():
132
  with st.sidebar:
133
  st.title("Model Selection")
134
 
135
- # Textbox for System Prompt
136
- st.markdown("### System Prompt")
137
- system_prompt = st.text_area(
138
- "Enter System Prompt:",
139
- value=st.session_state.system_prompt,
140
- height=100
141
- )
142
- st.session_state.system_prompt = system_prompt # Update the session state
143
 
144
  selected_model = st.selectbox(
145
  "Choose a model:",
@@ -168,12 +167,12 @@ with st.sidebar:
168
  for conv_id, conv_data in st.session_state.conversations.items():
169
  if st.button(conv_data["title"], key=f"hist_{conv_id}"):
170
  st.session_state.current_conversation_id = conv_id
171
- st.rerun()
172
 
173
  # New Chat Button
174
  if st.button("+ New Chat"):
175
  create_new_chat()
176
- st.rerun()
177
 
178
  # Main chat window
179
  with st.container():
@@ -201,4 +200,13 @@ with st.container():
201
  full_response = get_ai_response(prompt, messages[:-1])
202
  st.markdown(full_response)
203
  messages.append({"role": "assistant", "content": full_response})
204
- st.rerun()
 
 
 
 
 
 
 
 
 
 
4
  import os
5
  import uuid
6
  from datetime import datetime
 
7
 
8
  # Load environment variables
9
  load_dotenv()
 
131
  with st.sidebar:
132
  st.title("Model Selection")
133
 
134
+ # Collapsible System Prompt Box
135
+ with st.expander("System Prompt", expanded=False):
136
+ system_prompt = st.text_area(
137
+ "Enter System Prompt:",
138
+ value=st.session_state.system_prompt,
139
+ height=100
140
+ )
141
+ st.session_state.system_prompt = system_prompt # Update the session state
142
 
143
  selected_model = st.selectbox(
144
  "Choose a model:",
 
167
  for conv_id, conv_data in st.session_state.conversations.items():
168
  if st.button(conv_data["title"], key=f"hist_{conv_id}"):
169
  st.session_state.current_conversation_id = conv_id
170
+ st.experimental_rerun()
171
 
172
  # New Chat Button
173
  if st.button("+ New Chat"):
174
  create_new_chat()
175
+ st.experimental_rerun()
176
 
177
  # Main chat window
178
  with st.container():
 
200
  full_response = get_ai_response(prompt, messages[:-1])
201
  st.markdown(full_response)
202
  messages.append({"role": "assistant", "content": full_response})
203
+ st.experimental_rerun()
204
+
205
+ # JavaScript for Auto-Scroll
206
+ scroll_script = """
207
+ <script>
208
+ var chatContainer = window.parent.document.querySelector('.main');
209
+ chatContainer.scrollTop = chatContainer.scrollHeight;
210
+ </script>
211
+ """
212
+ st.markdown(scroll_script, unsafe_allow_html=True)