Update app.py
Browse files
app.py
CHANGED
|
@@ -47,14 +47,14 @@ def add_watermark(image, text="gooditem gooditem gooditem gooditem gooditem"):
|
|
| 47 |
return combined
|
| 48 |
|
| 49 |
|
|
|
|
|
|
|
|
|
|
| 50 |
|
| 51 |
-
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
-
return match.group(1).strip()
|
| 56 |
-
else:
|
| 57 |
-
return None
|
| 58 |
|
| 59 |
#제품명 추출 함수 수정 (여러 모델명 처리)
|
| 60 |
|
|
|
|
| 47 |
return combined
|
| 48 |
|
| 49 |
|
| 50 |
+
def extract_info_block(article_text):
|
| 51 |
+
pattern = r'(KBS2.*?<모델명>.*?)(?:\n\n|$)'
|
| 52 |
+
match = re.search(pattern, article_text, re.DOTALL)
|
| 53 |
|
| 54 |
+
if match:
|
| 55 |
+
return match.group(1).strip()
|
| 56 |
+
else:
|
| 57 |
+
return None
|
|
|
|
|
|
|
|
|
|
| 58 |
|
| 59 |
#제품명 추출 함수 수정 (여러 모델명 처리)
|
| 60 |
|