Update app.py

app.py

@@ -12,7 +12,7 @@ def save_binary_file(file_name, data):
     f.write(data)
     f.close()
 
-def generate_image(prompt, image=None): # Removed output_filename parameter
+def generate_image(prompt, image=None):
     # Initialize client with the API key
     client = genai.Client(
         api_key="AIzaSyAQcy3LfrkMy6DqS_8MqftAXu1Bx_ov_E8",
@@ -89,21 +89,17 @@ def generate_image(prompt, image=None): # Removed output_filename parameter
 
 
 # Function to handle chat interaction
-def chat_handler(prompt, user_image, chat_history): # Removed output_filename parameter
+def chat_handler(prompt, user_image, chat_history):
     # Add the user prompt to the chat history - ONLY TEXT PROMPT for user message
     if prompt:
         chat_history.append({"role": "user", "content": prompt})
     if user_image is not None:
-        # If there's a user image, add a separate message for the *thumbnail* to chat history
-        user_thumbnail_size = (100, 100)
-        user_thumbnail = user_image.copy()
-        user_thumbnail.thumbnail(user_thumbnail_size)
+        # If there's a user image, add a separate message for the high-quality image in a smaller container
         buffered = io.BytesIO()
-        user_thumbnail.save(buffered, format="PNG")
-        user_thumbnail_base64 = base64.b64encode(buffered.getvalue()).decode()
-        user_thumbnail_data_uri = f"data:image/png;base64,{user_thumbnail_base64}"
-        chat_history.append({"role": "user", "content": gr.HTML(f'<img src="{user_thumbnail_data_uri}" alt="Uploaded Image Thumbnail">')})
-
+        user_image.save(buffered, format="PNG")
+        user_image_base64 = base64.b64encode(buffered.getvalue()).decode()
+        user_image_data_uri = f"data:image/png;base64,{user_image_base64}"
+        chat_history.append({"role": "user", "content": gr.HTML(f'<img src="{user_image_data_uri}" alt="Uploaded Image" style="width:100px; height:100px; object-fit:contain;">')})
 
     # If no input, return early
     if not prompt and not user_image:
@@ -111,27 +107,21 @@ def chat_handler(prompt, user_image, chat_history): # Removed output_filename pa
         return chat_history, user_image, None, ""
 
     # Generate image based on user input
-    img, status = generate_image(prompt or "Generate an image", user_image) # Removed output_filename argument
+    img, status = generate_image(prompt or "Generate an image", user_image)
 
     thumbnail_data_uri = None # Initialize to None
     if img:
-        # Create thumbnail for chatbot
-        thumbnail_size = (100, 100) # Define thumbnail size
-        thumbnail = img.copy()
-        thumbnail = thumbnail.convert("RGB") # Force RGB mode for thumbnail
-        thumbnail.thumbnail(thumbnail_size)
-
-        # Convert thumbnail to base64 for chatbot display
+        # Use full-resolution image in a smaller container
+        img = img.convert("RGB") # Force RGB mode for consistency
         buffered = io.BytesIO()
-        thumbnail.save(buffered, format="PNG")
+        img.save(buffered, format="PNG")
         thumbnail_base64 = base64.b64encode(buffered.getvalue()).decode()
         thumbnail_data_uri = f"data:image/png;base64,{thumbnail_base64}"
-        print("Thumbnail Data URI:", thumbnail_data_uri) # Print to console
-        assistant_message_content = gr.HTML(f'<img src="{thumbnail_data_uri}" alt="Generated Image Thumbnail">') # Use gr.HTML
+        print("Image Data URI:", thumbnail_data_uri) # Print to console
+        assistant_message_content = gr.HTML(f'<img src="{thumbnail_data_uri}" alt="Generated Image" style="width:100px; height:100px; object-fit:contain;">') # Use gr.HTML with CSS
     else:
         assistant_message_content = status # If no image, send text status
 
-
     # Update chat history - Assistant message is now EITHER gr.HTML or text
     chat_history.append({"role": "assistant", "content": assistant_message_content})
 
@@ -169,12 +159,6 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
                 placeholder="Enter your image description here...",
                 lines=3
             )
-            # Removed filename_input
-            # filename_input = gr.Textbox(
-            #     label="Output Filename",
-            #     value="generated_image",
-            #     placeholder="Enter desired filename (without extension)"
-            # )
             generate_btn = gr.Button("Generate Image")
 
     # State to maintain chat history
@@ -183,14 +167,14 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
     # Connect the button to the chat handler
     generate_btn.click(
         fn=chat_handler,
-        inputs=[prompt_input, image_input, chat_state], # Removed filename_input
+        inputs=[prompt_input, image_input, chat_state],
         outputs=[chatbot, uploaded_image_output, generated_image_output, prompt_input]
     )
 
     # Also allow Enter key to submit
     prompt_input.submit(
         fn=chat_handler,
-        inputs=[prompt_input, image_input, chat_state], # Removed filename_input
+        inputs=[prompt_input, image_input, chat_state],
         outputs=[chatbot, uploaded_image_output, generated_image_output, prompt_input]
     )