Update app.py

app.py

@@ -54,24 +54,19 @@ def extract_info_block(article_text):
     else:
         return None
 
+# 브랜드/모델명 여러 개 추출
 def extract_product_info(article_text):
-    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)
+    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 []
 
     result = []
 
     if brand and models:
         brand_words = re.findall(r'[A-Za-z]+', brand)
-        selected_brand = ' '.join(brand_words[:2])
+        selected_brand = ' '.join(brand_words[:2])  # 앞 2단어만
 
         for model in models:
             search_query = f"{selected_brand} {model}"