ankban commited on
Commit
001a676
·
verified ·
1 Parent(s): 4121d13

Update app_utils.py

Browse files
Files changed (1) hide show
  1. 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
- client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
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
- # === SQLModel setup ===
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
- engine = create_engine(f"sqlite:///{db_path}")
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):