Update app.py

app.py

@@ -23,12 +23,12 @@ import time
 # import nest_asyncio
 load_dotenv()
 GROQ_API_KEY = os.getenv('GROQ_API_KEY')
-print(GROQ_API_KEY)
+# print(GROQ_API_KEY)
 # llm = ChatGroq(temperature=0, groq_api_key=GROQ_API_KEY, model_name="mixtral-8x7b-32768")
 client = Groq(api_key=GROQ_API_KEY)
 USER_AVATAR = "👤"
 BOT_AVATAR = "🤖"
-import asyncio
+# import asyncio
 
 st.set_page_config(layout="wide")
 
@@ -316,9 +316,11 @@ def upload():
         #     qa = rag("\n\n".join(json.dumps(data_dict) for data_dict in full_string))
         # else:
         if not st.session_state['non_form_inference_performed']:
-            qa = rag("\n\n".join(json.dumps(data_dict) for data_dict in st.session_state['non_form_inference_data']))
-            st.session_state['non_form_inference_performed'] = True
-            st.session_state.rag = qa
+            with st.spinner('Please wait for RAG-Setup to Complete...'):
+                qa = rag("\n\n".join(json.dumps(data_dict) for data_dict in st.session_state['non_form_inference_data']))
+                st.session_state['non_form_inference_performed'] = True
+                st.session_state.rag = qa
+                st.success('Done!')
        
         # col1, col2, col3 = st.columns([4,1,4]) 
         with col2: