modified back mGE STEP 1

app.py

@@ -10,6 +10,8 @@ import ast
 from server import *
 import cv2
 from typing import Optional
+import base64
+
 
 #http://localhost:8000
 app = FastAPI()
@@ -38,15 +40,37 @@ async def image_step1(image_file: UploadFile = File(...),background_image: Optio
         image = Image.open(io.BytesIO(contents_img))
         
 
-        produced_image=SegmenterBackground().Back_step1(image,input_to_type_of_filters,int(blur_radius))[0]#---->
+        output_step1=SegmenterBackground().Back_step1(image,input_to_type_of_filters,int(blur_radius))
+        produced_image=output_step1[0]
 
         # Save the processed image to a temporary file
-        output_file_path_tmp = "/tmp/tmp_processed_image.png"
-        produced_image.save(output_file_path_tmp)
+        #output_file_path_tmp = "/tmp/tmp_processed_image.png"
+        #produced_image.save(output_file_path_tmp)
 
         # Return the processed image for download
-        return FileResponse(output_file_path_tmp, media_type='image/png', filename="/tmp/tmp_processed_image.png")
-    
+        # return FileResponse(output_file_path_tmp, media_type='image/png', filename="/tmp/tmp_processed_image.png")
+
+        '''
+        # Convert the image to base64 to return it in the response
+        buffered = io.BytesIO()
+        #image.save(buffered, format="PNG")
+        encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
+            
+        # Returning both text and the base64 image
+        return {
+                "filename": background_image.filename,
+                "message": "Image processed successfully",
+                "image_base64": encoded_img
+        }
+        '''
+
+        buffer = io.BytesIO()
+        produced_image.save(buffer, format="PNG")
+        buffer.seek(0)
+        return StreamingResponse(
+                buffer, media_type="image/png",
+                headers={"Content-Disposition": "attachment; filename=processed_image.png"}
+            )
 
 @app.post('/imageStep2')
 async def image_step2(image_file: UploadFile = File(...),things_replace: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):