Update app.py

app.py

@@ -23,7 +23,7 @@ def main():
 
     csv_files = st.file_uploader("Upload CSV files", type="csv", accept_multiple_files=True)
     if csv_files:
-        llm = OpenAI(api_key=api_key, temperature=0, max_tokens=500)  # Adjust max_tokens as needed
+        llm = OpenAI(api_key=api_key, temperature=0, max_tokens=1000, output_format="list")  # Adjust max_tokens as needed
         user_input = st.text_input("Question here:")
 
         # Iterate over each CSV file
@@ -36,15 +36,6 @@ def main():
                 # Perform any necessary data preprocessing or feature engineering here
                 # You can modify the code based on your specific requirements
 
-                # Example: Accessing columns from the DataFrame
-                # column_data = df["column_name"]
-
-                # Example: Applying transformations or calculations to the data
-                # transformed_data = column_data.apply(lambda x: x * 2)
-
-                # Example: Using the preprocessed data with the OpenAI API
-                # llm_response = llm.predict(transformed_data)
-
                 if user_input:
                     # Pass the user input to the OpenAI agent for processing
                     agent = create_csv_agent(llm, f.name, verbose=True)
@@ -52,7 +43,10 @@ def main():
 
                     st.write(f"CSV File: {csv_file.name}")
                     st.write("Response:")
-                    st.write(response)
+
+                    # Create a foldable section for long outputs
+                    with st.beta_expander("Show Full Response"):
+                        st.write(response)
 
 # Add links to the sidebar with the same spacing properties
     st.sidebar.markdown("<p class='sidebar-link'>📚  <a href='https://chandrakalagowda-demo2.hf.space/'>  PDF Bot </a></p>", unsafe_allow_html=True)