Update app.py

app.py

@@ -75,10 +75,18 @@ if prompt := st.chat_input("Ask me anything.."):
     
     # full_response = st.session_state.chain.invoke(prompt).split("\n<|")[0]
     full_response = st.session_state.chain.invoke({"question":prompt, "memory":st.session_state.memory, "df_example":format_df(st.session_state.df.head(4))}).split("\n<|")[0]
-    
 
+    print(len(full_response))
+    
+    try :
+        exec(full_response)
+        full_response = "Here is the python code: \n\n```python"+ full_response +"\n```\n\nGenerated Response: \n\n"+ str(response)
+    except:
+        pass
+        
     with st.chat_message("assistant"):
         st.markdown(full_response)
+    
 
     # Display assistant response in chat message container
     # with st.chat_message("assistant"):