Update src/streamlit_app.py

src/streamlit_app.py

@@ -204,27 +204,35 @@ text=st.text_area("Ask your question:", height=100)
 document_button=st.pills("Ask based on Documents", ['search'], selection_mode="single")
 generate_button=st.button("Generate Response")
 if generate_button:
-    if document_button:
-        graph=load_graph(st.session_state.embeddings,st.session_state.chunks)
-        graph=graph.compile()
-        initial_state = {
-            "embedded_query":embed_sentence(text),
-            "knowledge": [],
-            "summary": "",
-            "final_response": None,}
-        final_state = graph.invoke(initial_state)
-        updated_text = f"""
-        Then respond to the client. Also follow the retrived information in the ##Summary section.
-        ## Instructions:
-        {text}
-        ## Summary:
-       {final_state['summary']}
-        """
-        complete_payment()
-        response=respond_chat(updated_text)
+    with st.spinner("Generating code..."):
+        try:
+            if document_button:
+                graph=load_graph(st.session_state.embeddings,st.session_state.chunks)
+                graph=graph.compile()
+                initial_state = {
+                    "embedded_query":embed_sentence(text),
+                    "knowledge": [],
+                    "summary": "",
+                    "final_response": None,}
+                final_state = graph.invoke(initial_state)
+                updated_text = f"""
+                Then respond to the client. Also follow the retrived information in the ##Summary section.
+                ## Instructions:
+                {text}
+                ## Summary:
+            {final_state['summary']}
+                """
+                complete_payment()
+                if st.session_state.db_transaction:
+                    response=respond_chat(updated_text)
+
+            else: 
+                complete_payment()
+                if st.session_state.db_transaction:
+                    response=respond_chat(text)
+        except Exception as e:
+                    st.error(f"Error during code generation: {e}")
 
-    else: 
-        response=respond_chat(text)
     col1,col2=st.columns([2,1])
     with col2:
         st.write("### Thought Process")