update

pages/{3_explanation_generation.py → 2_explanation_generation.py}

@@ -5,7 +5,7 @@ import os
 
 
 # Function to generate explanations based on a template
-def generate_explanations(model_name, questions, template, custom_template=None):
+def generate_explanations(model_name, questions, template, temperature, max_tokens, custom_template=None):
     agent = GPTAgent(model_name)
     explanations = []
     progress_bar = st.progress(0)
@@ -30,7 +30,7 @@ def generate_explanations(model_name, questions, template, custom_template=None)
 
             Explanation:
             """
-        response = agent.invoke(prompt, temperature=0.8, max_tokens=150).strip()
+        response = agent.invoke(prompt, temperature=temperature, max_tokens=max_tokens).strip()
         explanations.append(response)
 
         # Update progress bar
@@ -106,6 +106,9 @@ else:
         template = st.selectbox("Select an explanation template", ["Default", "Chain of Thought", "Custom"])
         model_name = st.selectbox('Select a model:', ['gpt4-1106', 'gpt35-1106'])
 
+        temperature = st.sidebar.slider('Temperature', min_value=0.0, max_value=1.0, value=0.8)
+        max_tokens = st.sidebar.slider('Max Tokens', min_value=50, max_value=500, value=150)
+
         custom_template = ""
         if template == "Custom":
             custom_template = st.text_area("Enter your custom template",
@@ -114,7 +117,7 @@ else:
         if st.button('Generate Explanations'):
             questions_df = pd.read_csv(uploaded_file_gen)
             questions = questions_df['question'].tolist()
-            explanations = generate_explanations(model_name, questions, template, custom_template)
+            explanations = generate_explanations(model_name, questions, template, temperature, max_tokens, custom_template)
 
             result_df_gen = pd.DataFrame({
                 'question': questions,