Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import pandas as pd
 import logging
-#from IPython.display import Markdown, display
+from IPython.display import Markdown, display
 from llama_index.core.query_engine import PandasQueryEngine
 from llama_index.llms.anthropic import Anthropic
 
@@ -9,20 +9,20 @@ logging.basicConfig(level=logging.INFO)
 
 def main():
     st.title('CSV Data Query Tool')
-    
+
     # Upload CSV file
     uploaded_file = st.file_uploader("Upload CSV file", type=['csv'])
     if uploaded_file is not None:
         df = pd.read_csv(uploaded_file)
-        
+
         # Input API Key
         api_key = st.text_input("Enter your ANTHROPIC_API_KEY:")
-        
+
         # Initialize Anthropic
         if api_key:
             llm = Anthropic(temperature=0.0, model='claude-3-opus-20240229', api_key=api_key)
             query_engine = PandasQueryEngine(df=df, llm=llm, verbose=True)
-            
+
             # Input user question
             question = st.text_input("Ask a question about the data:")
             if st.button("Submit"):
@@ -30,5 +30,10 @@ def main():
                     response = query_engine.query(question)
                     st.markdown(f"**Response:** {response}")
 
+        # Clear input fields
+        if st.button("Clear"):
+            st.text_input("Enter your ANTHROPIC_API_KEY:", value='')
+            st.text_input("Ask a question about the data:", value='')
+
 if __name__ == "__main__":
     main()