spinner added

app.py

@@ -56,28 +56,29 @@ if prompt := st.chat_input("What is up?"):
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    if option == "Sentiment Analysis":
-        pipe_sentiment_analysis = pipeline("sentiment-analysis")
-        if prompt:
-            out = pipe_sentiment_analysis(prompt)
-            doc = f"""
-                Prompt: {prompt}
-                Sentiment: {out[0]["label"]}
-                Score: {out[0]["score"]}
-            """
-    elif option == "Medical Summarization":
-        pipe_summarization = pipeline(
-            "summarization", model="Falconsai/medical_summarization"
-        )
-        if prompt:
-            out = pipe_summarization(prompt)
-            doc = out[0]["summary_text"]
-    elif option == "ChatGPT":
-        if prompt:
-            out = call_chatgpt(query=prompt)
-            doc = out
-    else:
-        None
+    with st.spinner("Wait for it..."):
+        if option == "Sentiment Analysis":
+            pipe_sentiment_analysis = pipeline("sentiment-analysis")
+            if prompt:
+                out = pipe_sentiment_analysis(prompt)
+                doc = f"""
+                    Prompt: {prompt}
+                    Sentiment: {out[0]["label"]}
+                    Score: {out[0]["score"]}
+                """
+        elif option == "Medical Summarization":
+            pipe_summarization = pipeline(
+                "summarization", model="Falconsai/medical_summarization"
+            )
+            if prompt:
+                out = pipe_summarization(prompt)
+                doc = out[0]["summary_text"]
+        elif option == "ChatGPT":
+            if prompt:
+                out = call_chatgpt(query=prompt)
+                doc = out
+        else:
+            doc = ""
 
     response = f"{doc}"
     # Display assistant response in chat message container