Update app.py

app.py

@@ -17,7 +17,11 @@ from collections import deque
 from audio_recorder_streamlit import audio_recorder
 import streamlit as st
 
-st.session_state.model_choice = ''
+st.set_page_config(page_title="Chat Toolkit",layout="wide")
+
+menu = ["htm", "txt", "xlsx", "csv", "md", "py"]  #619
+choice = st.sidebar.selectbox("Output File Type:", menu)
+model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301', 'gpt-4'))
 
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
@@ -31,7 +35,7 @@ def chat_with_model(prompt, document_section):
     conversation.append({'role': 'user', 'content': prompt})
     if len(document_section)>0:
         conversation.append({'role': 'assistant', 'content': document_section})
-    response = openai.ChatCompletion.create(model=st.session_state.model_choice, messages=conversation)
+    response = openai.ChatCompletion.create(model=model_choice)
     return response
     #return response['choices'][0]['message']['content']
 
@@ -146,19 +150,9 @@ def main():
     api_key = st.text_input("Enter OpenAI API Key:", type='password')
     if api_key:
         openai.api_key = api_key
-
-        #sidebar
-        models_list = ['gpt-3.5-turbo', 'gpt-3.5-turbo-0301', 'gpt-4']
-        model_choice = st.sidebar.radio("Select Model:", models_list, index=0)
-        st.session_state.model_choice = model_choice
-        
-        collength, colupload = st.columns([2,3])  # adjust the ratio as needed
         
-        with colupload:
-            uploaded_file = st.sidebar.file_uploader("Add a file for context:", type=["xml", "json", "xlsx","csv","html", "htm", "md", "txt"])
-        with collength:
-            #max_length = 12000 - optimal for gpt35 turbo. 2x=24000 for gpt4.  8x=96000 for gpt4-32k.
-            max_length = st.sidebar.slider("File section length for large files", min_value=1000, max_value=128000, value=12000, step=1000)
+        uploaded_file = st.sidebar.file_uploader("Add a file for context:", type=["xml", "json", "xlsx","csv","html", "htm", "md", "txt"])
+        max_length = st.sidebar.slider("File section length for large files", min_value=1000, max_value=128000, value=12000, step=1000)
         
         document_sections = deque()
         document_responses = {}