Update app.py

app.py

@@ -1,7 +1,13 @@
-import gradio as gr
+import base64
+import dash
+from dash import dcc, html, Input, Output, State
+import dash_bootstrap_components as dbc
+from dash.exceptions import PreventUpdate
 import google.generativeai as genai
 import requests
 import logging
+import threading
+import io
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -21,6 +27,28 @@ extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, si
 cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face
 """
 
+app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
+
+app.layout = dbc.Container([
+    html.H1("Stability AI SD3.5 Large Turbo Image Generator with Google Gemini Prompt Enhancement", className="my-4"),
+    dbc.Card([
+        dbc.CardBody([
+            dbc.Input(id="google-api-key", type="password", placeholder="Enter Google AI API Key", className="mb-3"),
+            dbc.Input(id="stability-api-key", type="password", placeholder="Enter Stability AI API Key", className="mb-3"),
+            dbc.Textarea(id="prompt", placeholder="Enter your prompt", className="mb-3"),
+            dcc.Dropdown(id="style", options=[{"label": s, "value": s} for s in STYLES], placeholder="Select style", className="mb-3"),
+            dbc.Textarea(id="negative-prompt", value=DEFAULT_NEGATIVE_PROMPT, className="mb-3"),
+            dbc.Button("Generate Image", id="submit-btn", color="primary", className="mb-3"),
+        ])
+    ], className="mb-4"),
+    dbc.Card([
+        dbc.CardBody([
+            html.Img(id="image-output", className="img-fluid"),
+            html.Div(id="enhanced-prompt-output", className="mt-3"),
+        ])
+    ])
+], fluid=True)
+
 def enhance_prompt(google_api_key, prompt, style):
     genai.configure(api_key=google_api_key)
     model = genai.GenerativeModel("gemini-2.0-flash-lite")
@@ -96,37 +124,42 @@ def process_and_generate(google_api_key, stability_api_key, prompt, style, negat
     try:
         enhanced_prompt = enhance_prompt(google_api_key, prompt, style)
         image_bytes = generate_image(stability_api_key, enhanced_prompt, style, negative_prompt)
-        
-        # Save image to a file
-        with open("generated_image.jpeg", "wb") as f:
-            f.write(image_bytes)
-        
-        # Return the file path and enhanced prompt
-        return "generated_image.jpeg", enhanced_prompt
+        return image_bytes, enhanced_prompt
     except Exception as e:
         logging.error(f"Error in process_and_generate: {str(e)}")
-        return str(e), enhanced_prompt if 'enhanced_prompt' in locals() else "Error occurred before prompt enhancement"
+        return None, str(e)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Stability AI SD3.5 Large Turbo Image Generator with Google Gemini Prompt Enhancement")
+@app.callback(
+    [Output("image-output", "src"),
+     Output("enhanced-prompt-output", "children")],
+    [Input("submit-btn", "n_clicks")],
+    [State("google-api-key", "value"),
+     State("stability-api-key", "value"),
+     State("prompt", "value"),
+     State("style", "value"),
+     State("negative-prompt", "value")],
+    prevent_initial_call=True
+)
+def update_output(n_clicks, google_api_key, stability_api_key, prompt, style, negative_prompt):
+    if n_clicks is None:
+        raise PreventUpdate
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            google_api_key = gr.Textbox(label="Google AI API Key", type="password")
-            stability_api_key = gr.Textbox(label="Stability AI API Key", type="password")
-            prompt = gr.Textbox(label="Prompt")
-            style = gr.Dropdown(label="Style", choices=STYLES)
-            negative_prompt = gr.Textbox(label="Negative Prompt", value=DEFAULT_NEGATIVE_PROMPT)
-            submit_btn = gr.Button("Generate Image")
-        
-        with gr.Column(scale=1):
-            image_output = gr.Image(label="Generated Image", type="filepath")
-            enhanced_prompt_output = gr.Textbox(label="Enhanced Prompt")
+    def run_process():
+        image_bytes, enhanced_prompt = process_and_generate(google_api_key, stability_api_key, prompt, style, negative_prompt)
+        if image_bytes:
+            encoded_image = base64.b64encode(image_bytes).decode('ascii')
+            return f"data:image/jpeg;base64,{encoded_image}", f"Enhanced Prompt: {enhanced_prompt}"
+        else:
+            return "", f"Error: {enhanced_prompt}"
+    
+    # Run the process in a separate thread
+    thread = threading.Thread(target=run_process)
+    thread.start()
+    thread.join()  # Wait for the thread to complete
     
-    submit_btn.click(
-        process_and_generate,
-        inputs=[google_api_key, stability_api_key, prompt, style, negative_prompt],
-        outputs=[image_output, enhanced_prompt_output]
-    )
+    return run_process()
 
-demo.launch()
+if __name__ == '__main__':
+    print("Starting the Dash application...")
+    app.run(debug=True, host='0.0.0.0', port=7860)
+    print("Dash application has finished running.")