Update app.py

app.py

@@ -123,27 +123,9 @@ if "df" in st.session_state:
     if st.button("Run Query"):
         with st.spinner("Running query..."):
             try:
-                # Split query execution into smaller chunks if needed
-                max_rows = 1000
-                total_rows = len(df)
-                results = []
-
-                for start in range(0, total_rows, max_rows):
-                    chunk = df.iloc[start:start + max_rows]
-                    chunk.to_csv("chunk_data.csv", index=False)
-                    partial_agent = create_csv_agent(
-                        ChatOpenAI(temperature=0, model="gpt-4", api_key=os.getenv("OPENAI_API_KEY")),
-                        path=["chunk_data.csv"],
-                        verbose=True,
-                        agent_type=AgentType.OPENAI_FUNCTIONS,
-                        allow_dangerous_code=True
-                    )
-                    result = partial_agent.invoke(query)
-                    results.append(result)
-
+                result = csv_agent.invoke(query)
                 st.success("Query executed successfully!")
                 st.write("### Query Result:")
-                st.write("\n".join(results))
-
+                st.write(result)
             except Exception as e:
-                st.error(f"Error executing query: {e}")
+                st.error(f"Error executing query: {e}")