Update app.py

app.py

@@ -9,6 +9,29 @@ theme = gr.themes.Soft(
     font="Rubik"  # Make sure "Rubik" is a valid font
 )
 
+# Create Gradio blocks with custom CSS for enhanced UI
+with gr.Blocks(
+    css="""
+        .gradio-container {background-color: #f0f4f8; padding: 20px;}
+        h1, h2, h3, h4, h5, h6 {color: #4a5568;}
+        .chat-message {background-color: #edf2f7; border-radius: 8px; padding: 10px;}
+        .gradio-container .avatar-container {height: 40px; width: 40px; border-radius: 50%; overflow: hidden;}
+        #duplicate-button {margin: auto; color: white; background: #2d3748; border-radius: 100vh; padding: 8px 16px;}
+        .tab-nav-button {background-color: #e2e8f0; border-radius: 8px;}
+        .tab-nav-button.selected {background-color: #cbd5e0;}
+    """,
+) as chat:
+    gr.Markdown("### Please note that only one image can be created at a time. To generate multiple images, consider using an Image Generator tool.")
+    gr.ChatInterface(
+        fn=model_inference,
+        chatbot=chatbot,
+        examples=EXAMPLES,
+        multimodal=True,
+        cache_examples=False,
+        autofocus=False,
+        concurrency_limit=10,
+    )
+
 # Create Gradio blocks with custom CSS for enhanced UI
 with gr.Blocks(theme=theme, title="RocketGPT - AI-Powered Chatbot") as demo:
     # Define individual tabs