refact: files structure

WilliamSilveiraF · Oct 25, 2023 · 3435ad0 · 3435ad0
1 parent f2eaf87
commit 3435ad0
Show file tree

Hide file tree

Showing 8 changed files with 7 additions and 10 deletions.
diff --git a/api/routes/audio.py b/api/routes/audio.py
@@ -1,6 +1,6 @@
 from fastapi import APIRouter, UploadFile, File, HTTPException
 from pathlib import Path
-from api.services.transcribe_audio import transcribe_model_selection
+from services.transcribe_audio import transcribe_audio_content
 
 import os
 
@@ -23,7 +23,7 @@ async def upload_audio(file: UploadFile = File(...)):
         with open(path_to_audio, 'wb') as f:
             f.write(data)
 
-        audio_text = transcribe_model_selection(path_to_audio)
+        audio_text = transcribe_audio_content(path_to_audio)
 
         return {"filename": file.filename, "message": "File uploaded successfully!"}
     except Exception as e:

diff --git a/main.py b/main.py
@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from api.routes import audio
-from api.middlewares import cors_middleware
+from middlewares import cors_middleware
 
 app = FastAPI()
 

diff --git a/api/middlewares/__init__.py → middlewares/__init__.py b/api/middlewares/__init__.py → middlewares/__init__.py
diff --git a/api/middlewares/cors_middleware.py → middlewares/cors_middleware.py b/api/middlewares/cors_middleware.py → middlewares/cors_middleware.py
diff --git a/api/services/__init__.py → services/__init__.py b/api/services/__init__.py → services/__init__.py
diff --git a/api/services/generate_summary.py → services/generate_summary.py b/api/services/generate_summary.py → services/generate_summary.py
diff --git a/api/services/sentiment_analysis.py → services/sentiment_analysis.py b/api/services/sentiment_analysis.py → services/sentiment_analysis.py
diff --git a/api/services/transcribe_audio.py → services/transcribe_audio.py b/api/services/transcribe_audio.py → services/transcribe_audio.py
@@ -1,6 +1,6 @@
 from google.cloud import speech
 
-def transcribe_model_selection(speech_file: str) -> speech.RecognizeResponse:
+def transcribe_audio_content(speech_file: str) -> speech.RecognizeResponse:
 
     client = speech.SpeechClient()
 
@@ -9,7 +9,7 @@ def transcribe_model_selection(speech_file: str) -> speech.RecognizeResponse:
 
     audio = speech.RecognitionAudio(content=content)
 
-    config = speech.RecognitionConfig(
+    config = speech.RecognitionConfig( # TODO SUPPORT DIFFERENT TYPES OF AUDIO
         encoding=speech.RecognitionConfig.AudioEncoding.ENCODING_UNSPECIFIED,
         sample_rate_hertz=16000,
         language_code="en-US",
@@ -19,8 +19,5 @@ def transcribe_model_selection(speech_file: str) -> speech.RecognizeResponse:
 
     for i, result in enumerate(response.results):
         alternative = result.alternatives[0]
-        print("-" * 20)
-        print(f"First alternative of result {i}")
-        print(f"Transcript: {alternative.transcript}")
-
-    return response
+
+        return alternative.transcript