Update app.py

app.py

@@ -90,7 +90,6 @@ def generate_image(prompt, image=None):
     print("Full Text Response from Gemini:", full_text_response) # Debugging full text
     return None, full_text_response # Return full text response
 
-
 # Function to handle chat interaction
 def chat_handler(prompt, user_image, chat_history):
     # Add the user prompt to the chat history - ONLY TEXT PROMPT for user message
@@ -150,19 +149,22 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
 
     # Input area
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=2):  # Increased scale for better spacing
             image_input = gr.Image(
                 label="Upload Image",
                 type="pil",
                 scale=1,
-                height=100
+                height=150,  # Increased height for better visibility
+                container=True,  # Ensure the component has a container for padding
+                elem_classes="p-4"  # Add padding via CSS class (4 units of padding)
             )
             prompt_input = gr.Textbox(
                 label="Prompt",
                 placeholder="Enter your image description here...",
-                lines=3
+                lines=3,
+                elem_classes="mt-2"  # Add margin-top for spacing between components
             )
-            generate_btn = gr.Button("Generate Image")
+            generate_btn = gr.Button("Generate Image", elem_classes="mt-2")  # Add margin-top for the button
 
     # State to maintain chat history
     chat_state = gr.State([])