Update merged_app2.py

merged_app2.py

@@ -759,8 +759,8 @@ def process(input_fg, prompt, image_width, image_height, num_samples, seed, step
 
     bg_source = BGSource(bg_source)
     
-    if bg_source == BGSource.NONE:
-        pass
+    # if bg_source == BGSource.NONE:
+    #     pass
     if bg_source == BGSource.UPLOAD:
         pass
     elif bg_source == BGSource.UPLOAD_FLIP:
@@ -926,7 +926,7 @@ quick_subjects = [[x] for x in quick_subjects]
 class BGSource(Enum):
     UPLOAD = "Use Background Image"
     UPLOAD_FLIP = "Use Flipped Background Image"
-    NONE = "None"
+    # NONE = "None"
     LEFT = "Left Light"
     RIGHT = "Right Light"
     TOP = "Top Light"
@@ -1050,7 +1050,7 @@ def use_orientation(selected_image:gr.SelectData):
 
 
 
-def generate_description(object_description,image, detail="high", max_tokens=250):
+def generate_description(object_description,image, detail="high", max_tokens=75):
     openai_api_key = os.getenv("OPENAI_API_KEY")
     client = OpenAI(api_key=openai_api_key)
 
@@ -1062,7 +1062,7 @@ def generate_description(object_description,image, detail="high", max_tokens=250
             img.save(buffered, format=IMAGE_FORMAT)
             img_base64 = base64.b64encode(buffered.getvalue()).decode()
 
-            prompt = f"As if you were describing the interior design, make a detailed caption of this image in one large paragraph. Highlighting textures, furnitures, locations. This object should be included in the description :{object_description}"
+            prompt = f"As if you were describing the interior design, make a detailed caption of this image in one paragraph. Highlighting textures, furnitures, locations. This object should be included in the description :{object_description}"
 
             payload = {
                 "model": "gpt-4o-mini",
@@ -1917,19 +1917,19 @@ with gr.Blocks() as app:
                 extracted_fg,  # Already processed RGBA image
             ],
             outputs=[orientation_result],
-        ).then(clear_memory, inputs=[], outputs=[])
+        )#.then(clear_memory, inputs=[], outputs=[])
         
         find_objects_button.click(
             fn=process_image,
             inputs=[input_fg, text_prompt],
             outputs=[extracted_objects, extracted_fg]
-            ).then(clear_memory, inputs=[], outputs=[])
+            )#.then(clear_memory, inputs=[], outputs=[])
         
         extract_button.click(
             fn=extract_foreground,
             inputs=[input_fg],
             outputs=[extracted_fg, x_slider, y_slider]
-        ).then(clear_memory, inputs=[], outputs=[])
+        )#.then(clear_memory, inputs=[], outputs=[])
         
     with gr.Tab("Style Transfer", visible=False):
         gr.Markdown("## Apply the style of an image to another one")
@@ -1984,7 +1984,7 @@ with gr.Blocks() as app:
 
         send_to_relight.click(move_prompt, [description_text], [prompt]).then(move_prompt, [description_text], [prompt_fill])
         
-        transfer_btn.click(send_img, [output_image], [input_fg]).then(clear_memory, inputs=[], outputs=[])
+        transfer_btn.click(send_img, [output_image], [input_fg])#.then(clear_memory, inputs=[], outputs=[])
 
         # describe_button.click(describe_image, [image_to_describe], [description_text])
 
@@ -1998,7 +1998,7 @@ with gr.Blocks() as app:
             fn=generate_image,
             inputs=[prompt_input, structure_image, style_image, depth_strength, style_strength],
             outputs=[output_image]
-        ).then(clear_memory, inputs=[], outputs=[])
+        )#.then(clear_memory, inputs=[], outputs=[])
 
 if __name__ == "__main__":
     app.queue(default_concurrency_limit=3)