Update app.py

app.py

@@ -54,19 +54,24 @@ def extract_info_block(article_text):
     else:
         return None
 
-# 브랜드/모델명 여러 개 추출
 def extract_product_info(article_text):
-    brand_match = re.search(r'<브랜드>[ \t]*([^\n]+)', article_text)
-    model_matches = re.findall(r'<모델명>[ \t]*([^\n]+)', article_text)
-
-    brand = brand_match.group(1).strip() if brand_match else None
-    models = [m.strip() for m in model_matches] if model_matches else []
+    brand = None
+    models = []
+
+    lines = article_text.splitlines()
+    for line in lines:
+        line = line.strip()
+        if line.startswith('<브랜드>'):
+            brand = line.replace('<브랜드>', '').strip()
+        elif line.startswith('<모델명>'):
+            model = line.replace('<모델명>', '').strip()
+            models.append(model)
 
     result = []
 
     if brand and models:
         brand_words = re.findall(r'[A-Za-z]+', brand)
-        selected_brand = ' '.join(brand_words[:2])  # 앞 2단어만
+        selected_brand = ' '.join(brand_words[:2])
 
         for model in models:
             search_query = f"{selected_brand} {model}"