Update app.py

app.py

@@ -21,6 +21,7 @@ st.set_page_config(
 menu = ["txt", "htm", "md", "py"]
 choice = st.sidebar.selectbox("Output file type:", menu)
 choicePrefix = "Output file type is "
+
 if choice == "txt":
      st.sidebar.write(choicePrefix + "Text File.")
 elif choice == "htm":
@@ -30,6 +31,16 @@ elif choice == "md":
 elif choice == "py":
      st.sidebar.write(choicePrefix + "Python Code.")
 
+model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-4-32k'))
+
+def chat_with_model(prompt, document_section):
+    model = model_choice
+    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
+    conversation.append({'role': 'user', 'content': prompt})
+    conversation.append({'role': 'assistant', 'content': document_section})
+    response = openai.ChatCompletion.create(model=model, messages=conversation)
+    return response['choices'][0]['message']['content']
+
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -53,14 +64,6 @@ def truncate_document(document, length):
 def divide_document(document, max_length):
     return [document[i:i+max_length] for i in range(0, len(document), max_length)]
 
-def chat_with_model(prompt, document_section):
-    model = "gpt-3.5-turbo"
-    conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
-    conversation.append({'role': 'user', 'content': prompt})
-    conversation.append({'role': 'assistant', 'content': document_section})
-    response = openai.ChatCompletion.create(model=model, messages=conversation)
-    return response['choices'][0]['message']['content']
-
 def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
         data = file.read()