Update app.py

app.py

@@ -28,6 +28,7 @@ if uploaded_file is not None:
     df1
     sdf = SmartDataframe(df1)
    
+container=st.empty
 prompt = st.text_area("Enter your prompt:")
 if st.button("Generate"):
 
@@ -37,9 +38,10 @@ if st.button("Generate"):
 
                 bar = st.progress(0)
                 #sdf = SmartDataframe(df1)
-                #df
-                #sdf.chat(prompt)
-                st.write(sdf.chat(prompt))            
+                container.empty
+                container.write(sdf.chat(prompt))
+                container.write(st.image('./exports/charts/temp_chart.png',caption=None))
+                #st.write(sdf.chat(prompt))            
                 bar.progress(100)
 
         else:
@@ -50,7 +52,7 @@ if st.button("Generate"):
 #agent = Agent(df)
 #result = agent.chat("erstelle balkendiagramm")
 #st.write(result)
-sdf = SmartDataframe(df)
+#sdf = SmartDataframe(df)
 #sdf.chat("draw chart")
 #pandas_ai = PandasAI(llm, verbose=True, save_charts=True)
 #st.write(sdf.chat("Plot a chart"))