From 7620fd667919b2a62f1d0627fc195100fe902a0b Mon Sep 17 00:00:00 2001
From: marioggil <marioggil@gmail.com>
Date: Tue, 10 Sep 2024 15:52:30 -0500
Subject: [PATCH] feat: Work with png and jpg

---
 apis.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/apis.py b/apis.py
index 86f8e18..52899c5 100644
--- a/apis.py
+++ b/apis.py
@@ -97,9 +97,16 @@ menuLLM="""        <option value="meta-llama/Meta-Llama-3.1-70B-Instruct">meta-l
 
 @app.post("/uploadimg")
 def upload_image(image: UploadFile = File(...),type="factura"):
-    endfile=".jpg"
+    if image.headers['content-type']=='image/png':
+        endfile=".png"
+    elif image.headers['content-type']=='image/jpeg':
+        endfile=".jpg"
+    else:
+        return JSONResponse(content={
+            "message": f"The file should be png or jpg"
+        }, status_code=500)
+    
     t=time.time()
-    print(image.headers)
     try:
         # Create a temporary file to store the uploaded audio
         headfilename=uuid.uuid4()