Update app.py

app.py

@@ -43,8 +43,15 @@ app.layout = dbc.Container([
     ], className="mb-4"),
     dbc.Card([
         dbc.CardBody([
-            html.Img(id="image-output", className="img-fluid"),
-            html.Div(id="enhanced-prompt-output", className="mt-3"),
+            dcc.Loading(
+                id="loading",
+                type="circle",
+                children=[
+                    html.Div(id="status-message", className="mb-3"),
+                    html.Img(id="image-output", className="img-fluid"),
+                    html.Div(id="enhanced-prompt-output", className="mt-3"),
+                ]
+            ),
         ])
     ])
 ], fluid=True)
@@ -120,18 +127,25 @@ def generate_image(stability_api_key, enhanced_prompt, style, negative_prompt):
         logging.error(f"Request failed: {str(e)}")
         raise Exception(f"Request failed: {str(e)}")
 
-def process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt):
+def process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt, set_status):
     try:
+        set_status("Enhancing prompt...")
         enhanced_prompt = enhance_prompt(google_api_key, prompt, style)
+        
+        set_status("Generating image...")
         image_bytes = generate_image(stability_api_key, enhanced_prompt, style, negative_prompt)
+        
+        set_status("Image generated successfully!")
         return image_bytes, enhanced_prompt
     except Exception as e:
         logging.error(f"Error in process_and_generate: {str(e)}")
+        set_status(f"Error: {str(e)}")
         return None, str(e)
 
 @app.callback(
     [Output("image-output", "src"),
-     Output("enhanced-prompt-output", "children")],
+     Output("enhanced-prompt-output", "children"),
+     Output("status-message", "children")],
     [Input("submit-btn", "n_clicks")],
     [State("google-api-key", "value"),
      State("stability-api-key", "value"),
@@ -144,13 +158,18 @@ def update_output(n_clicks, google_api_key, stability_api_key, prompt, style, ne
     if n_clicks is None:
         raise PreventUpdate
     
+    status = {"message": "Starting process..."}
+    
+    def set_status(message):
+        status["message"] = message
+    
     def run_process():
-        image_bytes, enhanced_prompt = process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt)
+        image_bytes, enhanced_prompt = process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt, set_status)
         if image_bytes:
             encoded_image = base64.b64encode(image_bytes).decode('ascii')
-            return f"data:image/jpeg;base64,{encoded_image}", f"Enhanced Prompt: {enhanced_prompt}"
+            return f"data:image/jpeg;base64,{encoded_image}", f"Enhanced Prompt: {enhanced_prompt}", status["message"]
         else:
-            return "", f"Error: {enhanced_prompt}"
+            return "", f"Error: {enhanced_prompt}", status["message"]
     
     # Run the process in a separate thread
     thread = threading.Thread(target=run_process)