Update src/streamlit_app.py

src/streamlit_app.py

@@ -204,31 +204,32 @@ 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:
-    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:
+    complete_payment()
+    if st.session_state.db_transaction:
+
+        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()
+                    
+                        response=respond_chat(updated_text)
+    
+                else:
                     response=respond_chat(text)
         except Exception as e:
                     st.error(f"Error during code generation: {e}")