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, output_filename="generated_image"):
+def generate_image(prompt, image=None): # Removed output_filename parameter
     # Initialize client with the API key
     client = genai.Client(
         api_key="AIzaSyAQcy3LfrkMy6DqS_8MqftAXu1Bx_ov_E8",
@@ -74,7 +74,7 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
         if chunk.candidates[0].content.parts[0].inline_data:
             inline_data = chunk.candidates[0].content.parts[0].inline_data
             file_extension = mimetypes.guess_extension(inline_data.mime_type)
-            filename = f"{output_filename}{file_extension}"
+            filename = f"generated_image{file_extension}" # Hardcoded filename
             save_binary_file(filename, inline_data.data)
 
             # Convert binary data to PIL Image
@@ -89,7 +89,7 @@ def generate_image(prompt, image=None, output_filename="generated_image"):
 
 
 # Function to handle chat interaction
-def chat_handler(prompt, user_image, chat_history, output_filename="generated_image"):
+def chat_handler(prompt, user_image, chat_history): # Removed output_filename parameter
     # Add the user prompt to the chat history - ONLY TEXT PROMPT for user message
     if prompt:
         chat_history.append({"role": "user", "content": prompt})
@@ -111,7 +111,7 @@ def chat_handler(prompt, user_image, chat_history, output_filename="generated_im
         return chat_history, user_image, None, ""
 
     # Generate image based on user input
-    img, status = generate_image(prompt or "Generate an image", user_image, output_filename)
+    img, status = generate_image(prompt or "Generate an image", user_image) # Removed output_filename argument
 
     thumbnail_data_uri = None # Initialize to None
     if img:
@@ -169,11 +169,12 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
                 placeholder="Enter your image description here...",
                 lines=3
             )
-            filename_input = gr.Textbox(
-                label="Output Filename",
-                value="generated_image",
-                placeholder="Enter desired filename (without extension)"
-            )
+            # 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
@@ -182,14 +183,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, filename_input],
+        inputs=[prompt_input, image_input, chat_state], # Removed filename_input
         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, filename_input],
+        inputs=[prompt_input, image_input, chat_state], # Removed filename_input
         outputs=[chatbot, uploaded_image_output, generated_image_output, prompt_input]
     )