Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import streamlit as st
|
2 |
import whisper
|
3 |
-
from
|
4 |
import googletrans
|
5 |
from gtts import gTTS
|
6 |
from io import BytesIO
|
@@ -156,7 +156,9 @@ def load_ocr_reader(languages):
|
|
156 |
return easyocr.Reader(languages if languages else ['en'])
|
157 |
|
158 |
whisper_model = load_whisper_model()
|
159 |
-
|
|
|
|
|
160 |
|
161 |
# Navigation
|
162 |
icons = {"Audio Transcription": "π€", "Image OCR": "π", "Help": "β"}
|
@@ -377,9 +379,10 @@ elif selected_tab == "Image OCR":
|
|
377 |
progress_bar = simulate_progress()
|
378 |
|
379 |
try:
|
380 |
-
translation = asyncio.run(translator.translate(
|
381 |
-
|
382 |
-
))
|
|
|
383 |
st.session_state.translated_ocr_text = translation.text
|
384 |
st.success("β
Translation complete!")
|
385 |
except Exception as e:
|
|
|
1 |
import streamlit as st
|
2 |
import whisper
|
3 |
+
from deep_translator import GoogleTranslator
|
4 |
import googletrans
|
5 |
from gtts import gTTS
|
6 |
from io import BytesIO
|
|
|
156 |
return easyocr.Reader(languages if languages else ['en'])
|
157 |
|
158 |
whisper_model = load_whisper_model()
|
159 |
+
def translate_text(text, target_language='en'):
|
160 |
+
translator = GoogleTranslator(source='auto', target=target_language)
|
161 |
+
return translator.translate(text)
|
162 |
|
163 |
# Navigation
|
164 |
icons = {"Audio Transcription": "π€", "Image OCR": "π", "Help": "β"}
|
|
|
379 |
progress_bar = simulate_progress()
|
380 |
|
381 |
try:
|
382 |
+
#translation = asyncio.run(translator.translate(
|
383 |
+
# st.session_state.extracted_text, dest=language
|
384 |
+
#))
|
385 |
+
translation = translate_text(st.session_state.extracted_text, target_language=language)
|
386 |
st.session_state.translated_ocr_text = translation.text
|
387 |
st.success("β
Translation complete!")
|
388 |
except Exception as e:
|