Update app.py

app.py

@@ -29,12 +29,13 @@ def generate_filename(prompt, file_type):
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:45]
     return f"{safe_date_time}_{safe_prompt}.{file_type}"
 
+TEMPERATURE = st.sidebar.slider("Adjust Creativity:", min_value=0.1, max_value=1.0, value=0.5, step=0.1)
 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)
+    response = openai.ChatCompletion.create(model=model, messages=conversation, temperature=TEMPERATURE)
     return response['choices'][0]['message']['content']
     
 def create_file(filename, prompt, response):
@@ -207,4 +208,5 @@ def main():
                 st.experimental_rerun()
 
 if __name__ == "__main__":
-    main()
+    main()
+