import display_gloss as dg import synonyms_preprocess as sp from NLP_Spacy_base_translator import NlpSpacyBaseTranslator from flask import Flask, render_template, Response, request, send_file import io import cv2 import numpy as np import os import requests from urllib.parse import quote, unquote import tempfile import re app = Flask(__name__, static_folder='static') app.config['TITLE'] = 'Sign Language Translate' nlp, dict_docs_spacy = sp.load_spacy_values() dataset, list_2000_tokens = dg.load_data() def find_quoted_words(text): """작은따옴표로 묶인 단어들을 찾는 함수""" # 따옴표가 없는 단어에 따옴표 추가 (예: 한국'을 '한국'으로) text = re.sub(r"(\w+)'", r"'\1'", text) return re.findall(r"'([^']*)'", text) def spell_out_word(word): """단어를 개별 알파벳으로 분리하는 함수""" return ' '.join(list(word.lower())) def translate_quoted_word(word): """따옴표 안의 단어를 개별적으로 번역""" try: url = "https://translate.googleapis.com/translate_a/single" params = { "client": "gtx", "sl": "ko", "tl": "en", "dt": "t", "q": word } response = requests.get(url, params=params) if response.status_code == 200: translated = response.json()[0][0][0].upper() return translated return word except Exception as e: print(f"Word translation error: {e}") return word def translate_korean_to_english(text): """전체 텍스트 번역 함수""" try: # 1. 따옴표로 묶인 단어 찾기 quoted_words = re.findall(r"'([^']*)'", text) # 2. 각 따옴표 단어 개별 번역 translated_words = {} for word in quoted_words: url = "https://translate.googleapis.com/translate_a/single" params = { "client": "gtx", "sl": "ko", "tl": "en", "dt": "t", "q": word } response = requests.get(url, params=params) if response.status_code == 200: translated = response.json()[0][0][0].upper() translated_words[word] = translated # 임시로 원본 텍스트의 따옴표 부분을 특수 마커로 대체 text = text.replace(f"'{word}'", "PROPER_NOUN_MARKER") # 3. 전체 문장 번역 params = { "client": "gtx", "sl": "ko", "tl": "en", "dt": "t", "q": text } response = requests.get(url, params=params) if response.status_code == 200: full_translation = response.json()[0][0][0] # 4. 번역된 문장에서 마커를 번역된 고유명사로 대체 for original, translated in translated_words.items(): full_translation = full_translation.replace("PROPER_NOUN_MARKER", f"'{translated}'", 1) return full_translation else: raise Exception(f"Translation API returned status code: {response.status_code}") except Exception as e: print(f"Translation error: {e}") return text @app.route('/translate/', methods=['POST']) def result(): if request.method == 'POST': input_text = request.form['inputSentence'].strip() if not input_text: return render_template('error.html', error="Please enter text to translate") try: # 번역 수행 english_text = translate_korean_to_english(input_text) if not english_text: raise Exception("Translation failed") # 따옴표로 묶인 단어 추출 (번역된 영어 텍스트에서) quoted_words = [word.strip("'") for word in re.findall(r"'([^']*)'", english_text)] # 번역된 텍스트에서 따옴표 제거하고 ASL 변환 clean_english = re.sub(r"'([^']*)'", r"\1", english_text) eng_to_asl_translator = NlpSpacyBaseTranslator(sentence=clean_english) generated_gloss = eng_to_asl_translator.translate_to_gloss() # 단어 처리 processed_gloss = [] words = generated_gloss.split() for word in words: word_upper = word.upper() if any(quoted.upper() == word_upper for quoted in quoted_words): # 고유명사인 경우 철자를 하나씩 분리 spelled_word = spell_out_word(word) processed_gloss.extend(['FINGERSPELL-START'] + spelled_word.split() + ['FINGERSPELL-END']) else: # 일반 단어는 기존 방식대로 처리 word_lower = word.lower() if word_lower.isalnum(): processed_gloss.append(word_lower) gloss_sentence_before_synonym = " ".join(processed_gloss) # 고유명사가 아닌 단어들만 동의어 처리 final_gloss = [] i = 0 while i < len(processed_gloss): if processed_gloss[i] == 'FINGERSPELL-START': final_gloss.extend(processed_gloss[i:i+2]) # START와 첫 글자 추가 i += 2 while i < len(processed_gloss) and processed_gloss[i] != 'FINGERSPELL-END': final_gloss.append(processed_gloss[i]) i += 1 if i < len(processed_gloss): final_gloss.append(processed_gloss[i]) # END 추가 i += 1 else: word = processed_gloss[i] final_gloss.append(sp.find_synonyms(word, nlp, dict_docs_spacy, list_2000_tokens)) i += 1 gloss_sentence_after_synonym = " ".join(final_gloss) return render_template('result.html', title=app.config['TITLE'], original_sentence=input_text, english_translation=english_text, gloss_sentence_before_synonym=gloss_sentence_before_synonym, gloss_sentence_after_synonym=gloss_sentence_after_synonym) except Exception as e: return render_template('error.html', error=f"Translation error: {str(e)}") def generate_complete_video(gloss_list, dataset, list_2000_tokens): try: frames = [] is_spelling = False for gloss in gloss_list: if gloss == 'FINGERSPELL-START': is_spelling = True continue elif gloss == 'FINGERSPELL-END': is_spelling = False continue for frame in dg.generate_video([gloss], dataset, list_2000_tokens): frame_data = frame.split(b'\r\n\r\n')[1] nparr = np.frombuffer(frame_data, np.uint8) img = cv2.imdecode(nparr, cv2.IMREAD_COLOR) frames.append(img) if not frames: raise Exception("No frames generated") height, width = frames[0].shape[:2] fourcc = cv2.VideoWriter_fourcc(*'mp4v') with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_file: temp_path = temp_file.name out = cv2.VideoWriter(temp_path, fourcc, 25, (width, height)) for frame in frames: out.write(frame) out.release() with open(temp_path, 'rb') as f: video_bytes = f.read() os.remove(temp_path) return video_bytes except Exception as e: print(f"Error generating video: {str(e)}") raise @app.route('/') def index(): return render_template('index.html', title=app.config['TITLE']) @app.route('/video_feed') def video_feed(): sentence = request.args.get('gloss_sentence_to_display', '') gloss_list = sentence.split() return Response(dg.generate_video(gloss_list, dataset, list_2000_tokens), mimetype='multipart/x-mixed-replace; boundary=frame') @app.route('/download_video/') def download_video(gloss_sentence): try: decoded_sentence = unquote(gloss_sentence) gloss_list = decoded_sentence.split() if not gloss_list: return "No gloss provided", 400 video_bytes = generate_complete_video(gloss_list, dataset, list_2000_tokens) if not video_bytes: return "Failed to generate video", 500 return send_file( io.BytesIO(video_bytes), mimetype='video/mp4', as_attachment=True, download_name='sign_language.mp4' ) except Exception as e: print(f"Download error: {str(e)}") return f"Error downloading video: {str(e)}", 500 if __name__ == "__main__": app.run(host="0.0.0.0", port=7860, debug=True)