Update app_utils.py
Browse files- app_utils.py +8 -11
app_utils.py
CHANGED
|
@@ -1,23 +1,22 @@
|
|
| 1 |
-
|
|
|
|
|
|
|
| 2 |
import os
|
| 3 |
import uuid
|
| 4 |
import subprocess
|
| 5 |
-
from datetime import datetime
|
| 6 |
-
from sqlmodel import SQLModel, Field, create_engine, Session, select
|
| 7 |
-
from typing import Optional
|
| 8 |
from faster_whisper import WhisperModel
|
| 9 |
-
from openai import OpenAI
|
| 10 |
|
| 11 |
# === Setup ===
|
| 12 |
db_path = "/tmp/chatter_sessions.db"
|
| 13 |
-
|
|
|
|
| 14 |
|
| 15 |
LANG_CODES = {
|
| 16 |
"English": "en", "Spanish": "es", "Hindi": "hi", "French": "fr", "German": "de",
|
| 17 |
"Arabic": "ar", "Chinese": "zh", "Portuguese": "pt", "Japanese": "ja", "Korean": "ko"
|
| 18 |
}
|
| 19 |
|
| 20 |
-
# ===
|
| 21 |
class SessionEntry(SQLModel, table=True):
|
| 22 |
id: Optional[int] = Field(default=None, primary_key=True)
|
| 23 |
user: str
|
|
@@ -26,9 +25,7 @@ class SessionEntry(SQLModel, table=True):
|
|
| 26 |
feedback: str
|
| 27 |
language: str
|
| 28 |
|
| 29 |
-
|
| 30 |
-
SQLModel.metadata.create_all(engine)
|
| 31 |
-
|
| 32 |
def save_to_db(user, transcript, feedback, language):
|
| 33 |
session = Session(engine)
|
| 34 |
entry = SessionEntry(
|
|
@@ -49,7 +46,7 @@ def fetch_user_sessions(user):
|
|
| 49 |
session.close()
|
| 50 |
return results
|
| 51 |
|
| 52 |
-
# === Whisper ===
|
| 53 |
model = WhisperModel("base", compute_type="int8")
|
| 54 |
|
| 55 |
def convert_to_wav(input_file):
|
|
|
|
| 1 |
+
from sqlmodel import SQLModel, Field, create_engine, Session, select
|
| 2 |
+
from datetime import datetime
|
| 3 |
+
from typing import Optional
|
| 4 |
import os
|
| 5 |
import uuid
|
| 6 |
import subprocess
|
|
|
|
|
|
|
|
|
|
| 7 |
from faster_whisper import WhisperModel
|
|
|
|
| 8 |
|
| 9 |
# === Setup ===
|
| 10 |
db_path = "/tmp/chatter_sessions.db"
|
| 11 |
+
engine = create_engine(f"sqlite:///{db_path}")
|
| 12 |
+
SQLModel.metadata.create_all(engine)
|
| 13 |
|
| 14 |
LANG_CODES = {
|
| 15 |
"English": "en", "Spanish": "es", "Hindi": "hi", "French": "fr", "German": "de",
|
| 16 |
"Arabic": "ar", "Chinese": "zh", "Portuguese": "pt", "Japanese": "ja", "Korean": "ko"
|
| 17 |
}
|
| 18 |
|
| 19 |
+
# === Spoken Session Table ===
|
| 20 |
class SessionEntry(SQLModel, table=True):
|
| 21 |
id: Optional[int] = Field(default=None, primary_key=True)
|
| 22 |
user: str
|
|
|
|
| 25 |
feedback: str
|
| 26 |
language: str
|
| 27 |
|
| 28 |
+
# === Spoken Session Utilities ===
|
|
|
|
|
|
|
| 29 |
def save_to_db(user, transcript, feedback, language):
|
| 30 |
session = Session(engine)
|
| 31 |
entry = SessionEntry(
|
|
|
|
| 46 |
session.close()
|
| 47 |
return results
|
| 48 |
|
| 49 |
+
# === Whisper Model ===
|
| 50 |
model = WhisperModel("base", compute_type="int8")
|
| 51 |
|
| 52 |
def convert_to_wav(input_file):
|