Update src/main.py

src/main.py

@@ -1,3 +1,4 @@
+# app.py
 import display_gloss as dg
 import synonyms_preprocess as sp
 from NLP_Spacy_base_translator import NlpSpacyBaseTranslator 
@@ -7,6 +8,8 @@ import cv2
 import numpy as np
 import os
 import requests
+from urllib.parse import quote, unquote
+import tempfile
 
 app = Flask(__name__, static_folder='static')
 app.config['TITLE'] = 'Sign Language Translate'
@@ -22,11 +25,10 @@ def translate_korean_to_english(text):
             "sl": "ko",
             "tl": "en",
             "dt": "t",
-            "q": text.strip()  # 입력 텍스트의 앞뒤 공백 제거
+            "q": text.strip()
         }
         response = requests.get(url, params=params)
         if response.status_code == 200:
-            # 전체 번역 결과를 하나의 문자열로 결합
             translated_text = ' '.join(item[0] for item in response.json()[0] if item[0])
             return translated_text
         else:
@@ -36,27 +38,37 @@ def translate_korean_to_english(text):
         return text
 
 def generate_complete_video(gloss_list, dataset, list_2000_tokens):
-    frames = []
-    for frame in dg.generate_video(gloss_list, 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)
-    
-    height, width = frames[0].shape[:2]
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
-    temp_path = os.path.join('/tmp', 'temp.mp4')
-    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
+    try:
+        frames = []
+        for frame in dg.generate_video(gloss_list, 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():
@@ -100,16 +112,29 @@ def video_feed():
     return Response(dg.generate_video(gloss_list, dataset, list_2000_tokens), 
                    mimetype='multipart/x-mixed-replace; boundary=frame')
 
-@app.route('/download_video/<gloss_sentence>')
+@app.route('/download_video/<path:gloss_sentence>')
 def download_video(gloss_sentence):
-    gloss_list = gloss_sentence.split()
-    video_bytes = generate_complete_video(gloss_list, dataset, list_2000_tokens)
-    return send_file(
-        io.BytesIO(video_bytes),
-        mimetype='video/mp4',
-        as_attachment=True,
-        download_name='sign_language.mp4'
-    )
+    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)