Update main.py

main.py

@@ -1,10 +1,17 @@
+# import whisper
+
+# model = whisper.load_model("medium")
+# options = whisper.DecodingOptions(language="spanish", fp16=False)
+# result = model.transcribe("audio2.mp3", decode_options=options)
+# print(result["text"])
+
 import torch
 from transformers import pipeline
 
-from fastapi import BackgroundTasks, File, UploadFile, FastAPI
+from fastapi import BackgroundTasks, FastAPI
 from fastapi.responses import RedirectResponse
 
-from azure.storage.blob import BlobServiceClient
+from azure.storage.blob import BlobClient, ContentSettings
 
 STORAGEACCOUNTURL = "https://callreviewer.blob.core.windows.net"
 STORAGEACCOUNTKEY = "vXq0X89zOaQxQmv7UBGFqqa61V0FRE6Gx1TgJvbtxZn5zLJ1ETc9aGDbbotuSoQzf5ob9QTuXlof+AStdHXOpA=="
@@ -28,25 +35,23 @@ translate_token_id = all_special_ids[-6]
 
 app = FastAPI()
 
-def transcribe(msg):
+def transcribe_task():
     try:
-        print(msg)
         print("For processing...")
-        blob_service_client_instance = BlobServiceClient(account_url=STORAGEACCOUNTURL, credential=STORAGEACCOUNTKEY)
-        blob_client_instance = blob_service_client_instance.get_blob_client(CONTAINERNAME, BLOBNAME, snapshot=None)
-        with open('/home/user/data/audio.mp3', "wb") as file:
-            blob_data = blob_client_instance.download_blob()
-            file.write(blob_data.readall())
+        blob = BlobClient(account_url=STORAGEACCOUNTURL,container_name=CONTAINERNAME, blob_name=BLOBNAME,credential=STORAGEACCOUNTKEY)
+        blob.set_http_headers(ContentSettings(content_type='audio/mp3'))
+        with open('audio22.mp3', "wb") as file:
+            file.write(blob.download_blob().readall())
         pipe.model.config.forced_decoder_ids = [[2, transcribe_token_id]]
         print("Call pipeline...")
-        text = pipe('/home/user/data/audio.mp3', return_timestamps=True)
+        text = pipe('audio22.mp3', return_timestamps=True)
         print(text)
-    except error:
-        print(error)
+    except:
+        print("Error")
 
-@app.post("/transcribe")
+@app.get("/transcribe")
 async def transcribe(background_tasks: BackgroundTasks):
-    background_tasks.add_task(transcribe, msg="Hola")
+    background_tasks.add_task(transcribe_task)
     return {"text": "Processing file..."}
 
 @app.get("/text")