Update app.py

app.py

@@ -1,3 +1,32 @@
 import gradio as gr
 
-gr.load("models/black-forest-labs/FLUX.1-schnell").launch()
+# Load the model
+model = gr.load("models/black-forest-labs/FLUX.1-dev")
+
+# Define the chatbot function
+def chatbot(message, history):
+    # Add the user's message to the history
+    history.append({"role": "user", "content": message})
+    
+    # Get the response from the model
+    response = model(message)
+    
+    # Add the model's response to the history
+    history.append({"role": "assistant", "content": response})
+    
+    return history
+
+# Create the Gradio interface
+with gr.Blocks(css="footer {visibility: hidden}") as demo:
+    gr.Markdown("# FLUX.1-dev Chatbot")
+    gr.Markdown("Welcome! This chatbot uses the FLUX.1-dev model. Feel free to ask anything!")
+    
+    chatbot = gr.Chatbot(height=400)
+    msg = gr.Textbox(label="Type your message here...")
+    clear = gr.Button("Clear")
+
+    msg.submit(chatbot, [msg, chatbot], [chatbot])
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+# Launch the interface
+demo.launch()