archivartaunik commited on
Commit
1fc8bb5
·
verified ·
1 Parent(s): 6bbaae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -13
app.py CHANGED
@@ -5,8 +5,7 @@ import mimetypes
5
  from pydub import AudioSegment
6
 
7
  GEMINI_API_KEY = os.getenv("gemini")
8
- MODEL_NAME_TH = os.getenv("modTH")
9
- MODEL_NAME = os.getenv("mod")
10
 
11
  genai.configure(api_key=GEMINI_API_KEY)
12
 
@@ -18,26 +17,25 @@ def transcribe_audio(audio_file):
18
  with open(audio_file, "rb") as f:
19
  audio_data = f.read()
20
  prompt_text = (
21
- """The user wants me to transcribe the audio into subtitles in SRT format, with a maximum of three words per one subtitle.
22
- I need to listen to the audio and create subtitles with timestamps. Check time format hours:minutes:seconds,milliseconds (00:00:00,000) after creating all. Example SRT format:
23
  1
24
- 00:00:01,670 --> 00:00:02,030
25
  За мяжою, за мяжою,
26
-
27
  2
28
- 00:00:02,270 --> 00:00:03,850
29
  ні сваё і не чужое.
30
-
31
  3
32
- 00:00:04,240 --> 00:00:05,760
33
  Хоць спявай ты, хоць ты грай,
34
-
35
- 4
36
- 00:00:05,770 --> 00:00:06,250
37
  а навокал іншы край.
 
 
 
 
38
  """
39
  )
40
- model = genai.GenerativeModel(MODEL_NAME_TH)
41
  response = model.generate_content(
42
  [prompt_text, {"mime_type": mime_type, "data": audio_data}]
43
  )
 
5
  from pydub import AudioSegment
6
 
7
  GEMINI_API_KEY = os.getenv("gemini")
8
+ MODEL_NAME = os.getenv("modTH")
 
9
 
10
  genai.configure(api_key=GEMINI_API_KEY)
11
 
 
17
  with open(audio_file, "rb") as f:
18
  audio_data = f.read()
19
  prompt_text = (
20
+ """зрабі транскрыпцыю аўдыяфайла для стварэння субтытрыў з часовымі меткамі srt. Стварай субтытры карацей за 70 сімвалаў. Прыклад фармату часу 00:00:01,589. Пасля кожнага субтытра пакідай пусты радок.
21
+ Не пішы нічога, акрамя транскрыпцыі ў зададзеным фармаце! вось прыклад фармату:
22
  1
23
+ 00:00:15,670 --> 00:00:19,030
24
  За мяжою, за мяжою,
 
25
  2
26
+ 00:00:19,270 --> 00:00:22,850
27
  ні сваё і не чужое.
 
28
  3
29
+ 00:00:23,240 --> 00:00:29,760
30
  Хоць спявай ты, хоць ты грай,
 
 
 
31
  а навокал іншы край.
32
+ 4
33
+ 00:00:32,770 --> 00:00:39,250
34
+ Пакінуў я край родны,
35
+ спадзяюся, што не незваротна.
36
  """
37
  )
38
+ model = genai.GenerativeModel(MODEL_NAME)
39
  response = model.generate_content(
40
  [prompt_text, {"mime_type": mime_type, "data": audio_data}]
41
  )