Upload 2 files

app.py

@@ -1,60 +1,44 @@
 
 import gradio as gr
-from kiwipiepy import Kiwi
 from transformers import pipeline
-
-# 형태소 분석기
-kiwi = Kiwi()
-
-# 번역기 설정
-translators = {
-    "en": pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en"),
-    "zh": pipeline("translation", model="Helsinki-NLP/opus-mt-ko-zh"),
-    "vi": pipeline("translation", model="Helsinki-NLP/opus-mt-ko-vi")
-}
-
-# 문장 나누기
-def smart_split(text):
-    text = text.replace("다.", "다.★").replace("요.", "요.★").replace("니다.", "니다.★")
-    return [s.strip() for s in text.split("★") if s.strip()]
-
-# 핵심 요약 생성
-def disaster_style_summary(text):
-    parts = smart_split(text)
-    keywords = []
-
-    for part in parts:
-        tokens = kiwi.tokenize(part)
-        nouns = [t.form for t in tokens if t.tag.startswith("N") and len(t.form) > 1]
-        verbs = [t.form for t in tokens if t.tag.startswith("V") and len(t.form) > 1]
-        if any(v in part for v in ["바랍니다", "주의", "자제", "대피", "착용"]):
-            if nouns:
-                keywords.append(nouns[0] + " " + (verbs[0] if verbs else "주의"))
-        elif nouns:
-            keywords.append(nouns[0])
-    seen = []
-    clean = [k for k in keywords if not (k in seen or seen.append(k))]
-    return ". ".join(clean[:5])
-
-# 번역
-def translate(text, lang):
-    if lang == "ko":
-        return text
-    try:
-        result = translators[lang](text, max_length=100)
-        return result[0]["translation_text"]
-    except Exception as e:
-        return f"[번역 오류] {str(e)}"
-
-# 요약 + 번역 함수
+import re
+
+# 번역기 초기화 함수 (언어 선택 시 로딩)
+def load_translator(lang_code):
+    model_map = {
+        "en": "Helsinki-NLP/opus-mt-ko-en",
+        "zh": "Helsinki-NLP/opus-mt-ko-zh",
+        "vi": "Helsinki-NLP/opus-mt-ko-vi"
+    }
+    if lang_code == "ko":
+        return None
+    return pipeline("translation", model=model_map[lang_code])
+
+# 규칙 기반 요약
+def rule_based_summary(text):
+    key_phrases = re.findall(r"(폭염|호우|지진|감염병|불|화재|대피|주의|통제|착용|자제|신고|피해).*?(바랍니다|하세요|주의)", text)
+    flat = [" ".join(kp) for kp in key_phrases]
+    if not flat:
+        # 중요 문장이 없다면 명사 위주 추출
+        fallback = re.findall(r"[가-힣]{2,}", text)
+        return ". ".join(fallback[:3]) if fallback else "중요 문장 없음"
+    return ". ".join(flat[:3])
+
+# 전체 처리 함수
 def summarize_and_translate(text, lang):
     if not text.strip():
         return "입력된 내용이 없습니다.", ""
-    summary = disaster_style_summary(text)
-    translated = translate(summary, lang)
-    return summary, translated
+    summary = rule_based_summary(text)
+    if lang == "ko":
+        return summary, summary
+    try:
+        translator = load_translator(lang)
+        result = translator(summary, max_length=100)
+        return summary, result[0]["translation_text"]
+    except Exception as e:
+        return summary, f"[번역 오류] {str(e)}"
 
-# 인터페이스
+# Gradio UI
 iface = gr.Interface(
     fn=summarize_and_translate,
     inputs=[
@@ -65,7 +49,7 @@ iface = gr.Interface(
         gr.Textbox(label="📌 핵심 요약"),
         gr.Textbox(label="🌍 번역 요약")
     ],
-    title="📢 DISASTER SNAP v4"
+    title="📢 DISASTER SNAP v5 Lite"
 )
 
 iface.launch()

requirements.txt

@@ -1,5 +1,5 @@
+
 gradio
-kiwipiepy==0.16.1
 transformers
 sentencepiece
-torch
+torch