dockerupdatecjek

Dockerfile

@@ -1,6 +1,8 @@
-
 FROM python:3.9
 
+# Menjalankan perintah untuk menginstal paket-paket yang dibutuhkan
+RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6
+
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
@@ -9,4 +11,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -3,7 +3,7 @@ import pickle
 import uvicorn
 import pandas as pd
 import shutil
-# import cv2
+import cv2
 import tensorflow as tf
 import os
 from flask import Flask, jsonify, request, flash, redirect, url_for
@@ -89,25 +89,25 @@ def preprocessing(filepath):
 # ##                   API UNTUK MELAKUKAN PROSES PREDIKSI
 # ## -------------------------------------------------------------------------
 
-# @app.post('/upload/file',tags=["Predicting"])
-# async def upload_file(picture: UploadFile):
-#     file_extension = picture.filename.split('.')[-1].lower()
+@app.post('/upload/file',tags=["Predicting"])
+async def upload_file(picture: UploadFile):
+    file_extension = picture.filename.split('.')[-1].lower()
 
 
-#     if file_extension not in ALLOWED_EXTENSIONS:
-#         raise HTTPException(status_code=400, detail='Invalid file extension')
+    if file_extension not in ALLOWED_EXTENSIONS:
+        raise HTTPException(status_code=400, detail='Invalid file extension')
 
-#     os.makedirs(UPLOAD_FOLDER, exist_ok=True)
-#     file_path = os.path.join(UPLOAD_FOLDER, secure_filename(picture.filename))
-#     with open(file_path, 'wb') as f:
-#         f.write(picture.file.read())
-#     try:
-#         processed_img = preprocessing(cv2.imread(file_path))
-#     except Exception as e:
-#         os.remove(file_path)
-#         raise HTTPException(status_code=500, detail=f'Error processing image: {str(e)}')
+    os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+    file_path = os.path.join(UPLOAD_FOLDER, secure_filename(picture.filename))
+    with open(file_path, 'wb') as f:
+        f.write(picture.file.read())
+    try:
+        processed_img = preprocessing(cv2.imread(file_path))
+    except Exception as e:
+        os.remove(file_path)
+        raise HTTPException(status_code=500, detail=f'Error processing image: {str(e)}')
 
-#     return JSONResponse(content={'message': 'File successfully uploaded'}, status_code=200)
+    return JSONResponse(content={'message': 'File successfully uploaded'}, status_code=200)
 
 # @app.get('/get_images', tags=["Predicting"])
 # def get_images():