MohamedRashad commited on
Commit
8ddffb7
1 Parent(s): e9fd913

chore: Update app.py and requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +10 -10
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,10 +1,9 @@
1
  import tempfile
2
 
3
- import requests
4
  import edge_tts
5
  import gradio as gr
6
-
7
- tashkil_url = "http://www.7koko.com/api/tashkil/index.php"
8
 
9
  language_dict = {
10
  "English": {
@@ -443,20 +442,21 @@ language_dict = {
443
  }
444
  }
445
 
 
 
446
  async def text_to_speech_edge(text, language_code, speaker, tashkeel_checkbox=False):
447
 
 
448
  if language_code == "Arabic" and tashkeel_checkbox:
449
- data = {"textArabic": text}
450
- response = requests.post(tashkil_url, data=data)
451
- response.encoding = 'utf-8'
452
- text = response.text.strip()
453
  voice = language_dict[language_code][speaker]
454
  communicate = edge_tts.Communicate(text, voice)
455
  with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
456
- tmp_path = tmp_file.name
457
- await communicate.save(tmp_path)
458
 
459
- print(tmp_path)
460
  return text, tmp_path
461
 
462
 
 
1
  import tempfile
2
 
 
3
  import edge_tts
4
  import gradio as gr
5
+ from transformers import pipeline
6
+ import pyarabic.araby as araby
7
 
8
  language_dict = {
9
  "English": {
 
442
  }
443
  }
444
 
445
+ pipe = pipeline("text2text-generation", model="mush42/fine-tashkeel")
446
+
447
  async def text_to_speech_edge(text, language_code, speaker, tashkeel_checkbox=False):
448
 
449
+ # Remove diacritics from Arabic text then add tashkeel
450
  if language_code == "Arabic" and tashkeel_checkbox:
451
+ text = pipe(araby.strip_diacritics(text))[0]["generated_text"].strip()
452
+
453
+ # Get the voice for the selected language and speaker
 
454
  voice = language_dict[language_code][speaker]
455
  communicate = edge_tts.Communicate(text, voice)
456
  with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
457
+ tmp_path = tmp_file.name
458
+ await communicate.save(tmp_path)
459
 
 
460
  return text, tmp_path
461
 
462
 
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  gradio
2
  edge-tts
3
- requests
 
 
1
  gradio
2
  edge-tts
3
+ transformers
4
+ pyarabic