fix app

README.md

@@ -5,6 +5,7 @@ colorFrom: yellow
 colorTo: red
 sdk: gradio
 sdk_version: 4.38.1
+python_version: 3.9
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -5,7 +5,7 @@ from transformers import ReactCodeAgent, HfEngine, Tool
 import pandas as pd
 
 from gradio import Chatbot
-from streaming import stream_to_gradio
+from test_streaming import stream_to_gradio
 from huggingface_hub import login
 from gradio.data_classes import FileData
 
@@ -20,22 +20,18 @@ agent = ReactCodeAgent(
     max_iterations=10,
 )
 
-base_prompt = """You are an expert data analyst.
+base_prompt = """You are an expert full stack data analyst.
 You are given a data file and the data structure below.
 The data file is passed to you as the variable data_file, it is a pandas dataframe, you can use it directly.
 DO NOT try to load data_file, it is already a dataframe pre-loaded in your python interpreter!
-
+When plotting using matplotlib/seaborn save the figures to the (already existing) folder'./figures/': take care to clear each figure with plt.clf() before doing another plot.
+When filtering pandas dataframe use the iloc.
 When importing packages use this format: from package import module
 For example: from matplotlib import pyplot as plt
 Not: import matplotlib.pyplot as plt
 
-As you work, check for NoneType values and convert to NAN.
-
 Use the data file to answer the question or solve a problem given below.
 
-In your final answer: summarize your findings
-After each number derive real worlds insights, for instance: "Correlation between is_december and boredness is 1.3453, which suggest people are more bored in winter".
-
 Structure of the data:
 {structure_notes}
 

requirements.txt

@@ -4,6 +4,7 @@ seaborn
 scikit-learn
 scipy
 transformers
-pandas==2.2.2
+pandas
 huggingface_hub
-transformers
+numpy
+pandas