Krishna086 commited on
Commit
e44647f
·
verified ·
1 Parent(s): 689854b

Update lang_detect.py

Browse files
Files changed (1) hide show
  1. lang_detect.py +13 -13
lang_detect.py CHANGED
@@ -1,16 +1,16 @@
1
- from gtts import gTTS
2
- import io
3
 
4
- def text_to_speech(text, target_lang):
5
  try:
6
- if not text:
7
- return None
8
- lang_map = {"en": "en", "fr": "fr", "es": "es", "de": "de", "hi": "hi", "zh": "zh-cn", "ar": "ar", "ru": "ru", "ja": "ja"}
9
- lang_code = lang_map.get(target_lang, "en")
10
- tts = gTTS(text=text[:200], lang=lang_code)
11
- audio_buffer = io.BytesIO()
12
- tts.write_to_fp(audio_buffer)
13
- audio_buffer.seek(0)
14
- return audio_buffer if audio_buffer.getbuffer().nbytes > 0 else None
 
15
  except Exception:
16
- return None
 
1
+ from langdetect import detect_langs
 
2
 
3
+ def detect_language(text):
4
  try:
5
+ if len(text) < 10:
6
+ return [("English", 1.0, "English")]
7
+ lang_detections = detect_langs(text)
8
+ native_lang_map = {
9
+ "en": ("English", "English"), "fr": ("Français", "French"), "es": ("Español", "Spanish"),
10
+ "de": ("Deutsch", "German"), "hi": ("हिन्दी", "Hindi"), "zh": ("中文", "Chinese"),
11
+ "ar": ("العربية", "Arabic"), "ru": ("Русский", "Russian"), "ja": ("日本語", "Japanese")
12
+ }
13
+ detected_options = [(native_lang_map.get(lang.lang, ("English", "English"))[1], lang.prob, native_lang_map.get(lang.lang, ("English", "English"))[0]) for lang in lang_detections if lang.prob >= 0.7]
14
+ return detected_options[:3] if detected_options else [("English", 0.5, "English")]
15
  except Exception:
16
+ return [("English", 0.5, "English")]