Spaces:

taha454
/

Backg

Sleeping

App Files Files Community

LapStore commited on Sep 29, 2024

Commit

9beab52

1 Parent(s): 8791de8

modified video fast api

Browse files

Files changed (2) hide show

app.py +30 -9
index.html +16 -3

app.py CHANGED Viewed

@@ -65,13 +65,11 @@ async def image_step1(image_file: UploadFile = File(...),background_image: Optio
 @app.post('/imageStep2')
 async def image_step2(image_file: UploadFile = File(...),background_image: Optional [UploadFile] = File(None),type_of_filters: str = Form(...),
                       things_replace: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
-        contents = await image_file.read()
-        image = Image.open(io.BytesIO(contents))
         things_replace=ast.literal_eval(things_replace)
         blur_radius=int(blur_radius)
         input_to_type_of_filters=None
         if background_image and background_image.filename:
@@ -82,6 +80,10 @@ async def image_step2(image_file: UploadFile = File(...),background_image: Optio
                input_to_type_of_filters = type_of_filters
         produced_image=SegmenterBackground().Back_step2(image,input_to_type_of_filters,things_replace,int(blur_radius))
         # Save the processed image to a temporary file
@@ -92,18 +94,37 @@ async def image_step2(image_file: UploadFile = File(...),background_image: Optio
         return FileResponse(output_file_path_tmp, media_type='image/png', filename="/tmp/tmp_processed_image.png")
 @app.post('/Video')
-async def Video(video_file: UploadFile = File(...),kind_back: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
         #video_data = await video_file.read()
         #nparr = np.frombuffer(video_data, np.uint8)
         #video_path=cv2.imdecode(nparr, cv2.IMREAD_COLOR) #named this as just passed as it's path
-        video_path = f'/tmp/tmptmp.avi'#{video_file.filename}
-        with open(video_path, 'wb') as f:
             f.write(await video_file.read())
-        produced_video=SegmenterBackground().Back_video(video_path, '/tmp/29_sep_2.avi','cam',['animal','person'])#video,background_image,what_remove,blur_radius=23)
-        return StreamingResponse(open('/tmp/29_sep_2.avi', "rb"), media_type="video/avi")

 @app.post('/imageStep2')
 async def image_step2(image_file: UploadFile = File(...),background_image: Optional [UploadFile] = File(None),type_of_filters: str = Form(...),
                       things_replace: str = Form(...), blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
         things_replace=ast.literal_eval(things_replace)
         blur_radius=int(blur_radius)
         input_to_type_of_filters=None
         if background_image and background_image.filename:
                input_to_type_of_filters = type_of_filters
+        contents = await image_file.read()
+        image = Image.open(io.BytesIO(contents))
         produced_image=SegmenterBackground().Back_step2(image,input_to_type_of_filters,things_replace,int(blur_radius))
         # Save the processed image to a temporary file
         return FileResponse(output_file_path_tmp, media_type='image/png', filename="/tmp/tmp_processed_image.png")
 @app.post('/Video')
+async def Video(video_file: UploadFile = File(...),background_image: Optional [UploadFile] = File(None),kind_back: str = Form(...)
+                ,type_of_filters: str = Form(...),blur_radius: str = Form(...)):#--->,background_image: UploadFile = File(...)):
         #video_data = await video_file.read()
         #nparr = np.frombuffer(video_data, np.uint8)
         #video_path=cv2.imdecode(nparr, cv2.IMREAD_COLOR) #named this as just passed as it's path
+        blur_radius=int(blur_radius)
+        kind_back=ast.literal_eval(kind_back)
+        input_to_type_of_filters=None
+        if background_image and background_image.filename:
+                contents__back = await background_image.read()
+                image_back = Image.open(io.BytesIO(contents__back))
+                input_to_type_of_filters = image_back
+        else:
+               input_to_type_of_filters = type_of_filters
+        input_path_toWrite = f'/tmp/tmp_imput.avi'#{video_file.filename}
+        output_path = '/tmp/tmp_output.avi'
+        with open(input_path_toWrite, 'wb') as f:
             f.write(await video_file.read())
+        #--------> mp4?
+        SegmenterBackground().Back_video(input_path_toWrite, output_path,input_to_type_of_filters,kind_back,blur_radius)#video,background_image,what_remove,blur_radius=23)
+        return StreamingResponse(open(output_path, "rb"), media_type="video/avi")

index.html CHANGED Viewed

@@ -2,12 +2,12 @@
    <body bgcolor="#00cccc">
       <center>
          <br><br><br>
-         <form action="https://taha454-backg.hf.space/imageStep2" method="post"  enctype="multipart/form-data">
             <p><h3>Enter Image:</h3></p>
             type_of_filters :<p><input type="text" name="type_of_filters" /></p><br>
-            things_replace :<p><input type="text" name="things_replace" /></p><br>
             blur_radius :<p><input type="text" name="blur_radius" /></p><br>
-            Image : <input type="file" name="image_file" required><br><br>
             Background_Image : <input type="file" name="background_image" ><br><br>
             <p><input type="submit" value="submit" /></p>
@@ -15,3 +15,16 @@
       </center>
    </body>
 </html>

    <body bgcolor="#00cccc">
       <center>
          <br><br><br>
+         <form action="https://taha454-backg.hf.space/Video" method="post"  enctype="multipart/form-data">
             <p><h3>Enter Image:</h3></p>
             type_of_filters :<p><input type="text" name="type_of_filters" /></p><br>
+            kind_back :<p><input type="text" name="kind_back" /></p><br>
             blur_radius :<p><input type="text" name="blur_radius" /></p><br>
+            Video : <input type="file" name="image_file" required><br><br>
             Background_Image : <input type="file" name="background_image" ><br><br>
             <p><input type="submit" value="submit" /></p>
       </center>
    </body>
 </html>
+<!--
+<form action="https://taha454-backg.hf.space/imageStep2" method="post"  enctype="multipart/form-data">
+            <p><h3>Enter Image:</h3></p>
+            type_of_filters :<p><input type="text" name="type_of_filters" /></p><br>
+            things_replace :<p><input type="text" name="things_replace" /></p><br>
+            blur_radius :<p><input type="text" name="blur_radius" /></p><br>
+            Image : <input type="file" name="image_file" required><br><br>
+            Background_Image : <input type="file" name="background_image" ><br><br>
+            <p><input type="submit" value="submit" /></p>
+         </form>
+         -->