Update app.py

app.py

@@ -93,7 +93,7 @@ def read_email():
     except Exception as e:
         print(f"Error reading emails: {e}")
 
-def load_llm_model():
+async def load_llm_model():
     try:
         # Check if GPU is available and set the device accordingly
         device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -369,7 +369,7 @@ def change_bg_color(color):
         unsafe_allow_html=True
     )
 
-def start(running):
+async def start(running):
     while running:
         try:
             print('Started running...')
@@ -389,15 +389,17 @@ st.title("Streamlit LLM App")
 # Display buttons for starting and stopping the core function
 if st.button("Start"):
     st.write('Model loading...')  # Display a label when the model is loading
-    change_bg_color("yellow")  # Change background color to yellow
-    load_llm_model()  # Load the model
+    change_bg_color("lightyellow")  # Change background color to yellow
+    await load_llm_model()  # Load the model
+    change_bg_color("lightblue")
     st.success('Model loaded successfully!')
     if st.button('Analyse'):
+        change_bg_color('white')
         st.write('analyze started ! ...') # Show success message when model is loaded
-        start('True')
+        await start('True')
         change_bg_color("lightgreen")  # Change background color to green after loading
 
 if st.button("Stop"):
     st.write('Process stopped.')  # Display stop message
     change_bg_color("white")  # Reset background color
-    start('False')
+    await start('False')