Update main.py

main.py

@@ -1,8 +1,8 @@
 # main.py
+from deepmultilingualpunctuation import PunctuationModel
+import re
 
 from fastapi import FastAPI
-from PIL import Image
-import base64
 from fastapi.responses import HTMLResponse, FileResponse
 
 app = FastAPI()
@@ -30,24 +30,21 @@ async def root():
 
 @app.get("/api")
 async def cal_api():
-    images = []
-
-    with open('workdir/lion.jpg', 'rb') as open_file:
-        byte_content = open_file.read()
-    base64_bytes = base64.b64encode(byte_content)
-    base64_string = base64_bytes.decode('utf-8')
-    images.append(base64_string)
-
-    with open('workdir/cheetah.jpg', 'rb') as open_file:
-        byte_content = open_file.read()
-    base64_bytes = base64.b64encode(byte_content)
-    base64_string = base64_bytes.decode('utf-8')
-    images.append(base64_string)
-
-    #image_path='lion.jpg'
-    #pilim = Image.open(image_path)
-    #pilimrot = pilim.rotate(45)
-    return {"data": images}
+
+    input_text = "my name is clara i live in berkeley california"
+
+    model = PunctuationModel()
+    output_text = model.restore_punctuation(input_text)
+
+    split_text = output_text.split('. ')
+    pcnt_file_cr = '.\n'.join(split_text)
+    
+    regex1 = r"\bi\b"
+    regex2 = r"(?<=[.?!;])\s*\w"
+    regex3 = r"^\w"
+    pcnt_file_cr_cap = re.sub(regex3, lambda x: x.group().upper(), re.sub(regex2, lambda x: x.group().upper(), re.sub(regex1, "I", pcnt_file_cr)))
+
+    return {"data": pcnt_file_cr_cap}
 
 @app.get("/items/{item_id}")
 async def read_item(item_id):