Update app.py

app.py

@@ -638,7 +638,7 @@ def _():
 @app.cell
 def _(DirectLLMClient, mo):
     def display_response(user_prompt: str = None):
-        """Create a nice Marimo UI element to display the prompt and response."""
+        """Create a nice Marimo UI element to display the prompt and response with a loading spinner."""
         if not user_prompt.value:
             return mo.md(
                 f"""
@@ -657,7 +657,10 @@ def _(DirectLLMClient, mo):
 
         client = DirectLLMClient(provider='openai')
 
-        response = client.generate(prompt=user_prompt.value, system_message='You are a helpful assistant.')
+        # Display a loading message with a spinner
+        with mo.status.spinner(subtitle='Generating response...') as spinner:
+            response = client.generate(prompt=user_prompt.value, system_message='You are a helpful assistant.')
+            spinner.update(subtitle='Formatting response...')
 
         final_response = mo.vstack(
             [