Update app.py

app.py

@@ -168,50 +168,58 @@ def get_legal_context(query):
         return ""
     
     try:
-        relevant_info = []
-        
         # 쿼리 전처리
         query = query.strip()
         
         # 사건번호 형식 확인 (예: "2023다12345")
         is_case_number = any(char.isdigit() for char in query) and any(char.isalpha() for char in query)
         
-        for item in legal_dataset['train']:
-            case_number = item.get('사건번호', '')
-            
-            # 사건번호로 검색하는 경우 정확히 일치하는 것만 반환
-            if is_case_number:
+        if is_case_number:
+            # 사건번호로 정확한 검색
+            for item in legal_dataset['train']:
+                case_number = item.get('사건번호', '')
                 if query in case_number:
-                    case_info = (
-                        f"📌 사건번호: {case_number}\n"
-                        f"📅 선고일자: {item.get('선고일자', '')}\n"
-                        f"📝 판시사항:\n{item.get('판시사항', '')}\n\n"
-                        f"⚖️ 판결요지:\n{item.get('판결요지', '')}\n\n"
-                        f"📚 판례내용:\n{item.get('판례내용', '')}"
-                    )
-                    relevant_info.append(case_info)
-                    break  # 정확한 사건번호 매칭시 즉시 반환
-            else:
-                # 일반 검색의 경우 관련도 검사
-                searchable_text = f"{case_number} {item.get('판시사항', '')} {item.get('판결요지', '')}"
+                    # 전문 요청 확인
+                    if '전문' in query or '판례전문' in query:
+                        return (
+                            f"📜 판례 전문 (사건번호: {case_number})\n"
+                            f"━━━━━━━━━━━━━━━━━━━━━━━━━━━\n"
+                            f"{item.get('판례내용', '판례 전문을 찾을 수 없습니다.')}\n"
+                            f"━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+                        )
+                    else:
+                        # 기본 정보 반환
+                        return (
+                            f"📌 판례 정보 (사건번호: {case_number})\n"
+                            f"━━━━━━━━━━━━━━━━━━━━━━━━━━━\n"
+                            f"📅 선고일자: {item.get('선고일자', '')}\n"
+                            f"📝 판시사항:\n{item.get('판시사항', '')}\n\n"
+                            f"⚖️ 판결요지:\n{item.get('판결요지', '')}\n\n"
+                            f"💡 전문을 보시려면 '사건번호 전문'을 입력해주세요."
+                        )
+            return "❌ 해당하는 사건번호의 판례를 찾을 수 없습니다."
+        
+        else:
+            # 일반 키워드 검색
+            relevant_info = []
+            for item in legal_dataset['train']:
+                searchable_text = f"{item.get('사건번호', '')} {item.get('판시사항', '')} {item.get('판결요지', '')}"
                 if query.lower() in searchable_text.lower():
                     case_info = (
-                        f"📌 사건번호: {case_number}\n"
+                        f"📌 사건번호: {item.get('사건번호', '')}\n"
                         f"📅 선고일자: {item.get('선고일자', '')}\n"
-                        f"📝 판시사항:\n{item.get('판시사항', '')[:500]}...\n\n"
-                        f"⚖️ 판결요지:\n{item.get('판결요지', '')[:500]}..."
+                        f"📝 판시사항:\n{item.get('판시사항', '')[:300]}...\n\n"
+                        f"⚖️ 판결요지:\n{item.get('판결요지', '')[:300]}...\n"
+                        f"💡 전문을 보시려면 위 사건번호와 함께 '전문'을 입력해주세요."
                     )
                     relevant_info.append(case_info)
+                
+                if len(relevant_info) >= 3:
+                    break
             
-            if len(relevant_info) >= 3 and not is_case_number:  # 일반 검색의 경우만 여러 개 반환
-                break
-        
-        if relevant_info:
-            if is_case_number:
-                return "\n\n📜 판례 전문:\n" + relevant_info[0]
-            else:
-                return "\n\n관련 판례 정보:\n" + "\n\n---\n\n".join(relevant_info[:3])
-        return "❌ 해당하는 판례를 찾을 수 없습니다."
+            if relevant_info:
+                return "\n\n관련 판례 정보:\n" + "\n\n---\n\n".join(relevant_info)
+            return "❌ 검색어와 관련된 판례를 찾을 수 없습니다."
     
     except Exception as e:
         print(f"판례 검색 오류: {e}")