code added

Author: melihunsal

src/prompt_based/app.py

@@ -59,6 +59,8 @@ def generate_response(txt):
 
 example_submitted = False
 
+final_code_empty = st.empty()
+
 with st.form('a', clear_on_submit=True):
     submitted = st.form_submit_button('Submit')
     for col,example in zip(cols,examples):
@@ -92,7 +94,9 @@ def generate_response(txt):
                 if success:
                     bar.progress(75, text=PROGRESS_BAR_TEXTS["creating"])
                     example_submitted = False
-                    st.session_state['pid'] = streamlit_agent(demo_idea,demo_title,code,test_code,bar.progress,st.balloons)
+                    st.session_state['pid'], streamlit_code = streamlit_agent(demo_idea,demo_title,code,test_code,bar.progress,st.balloons)
+                    with st.expander("Code"):
+                        st.code(streamlit_code)
                     sleep(5)
                     webbrowser.open('http://localhost:8502')
 

src/prompt_based/model.py

@@ -286,4 +286,4 @@ def __call__(self,topic, title, code, test_code,progress_func,success_func):
         refined_code = self.refine_code(streamlit_code)
         progress_func(100,"Redirecting to the demo page...")
         success_func()
-        return self.run_code(refined_code)
+        return self.run_code(refined_code), refined_code