Update app.py

app.py

@@ -170,25 +170,52 @@ def get_legal_context(query):
     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']:
-            # 사건번호, 판시사항, 판결요지 등에서 검색
-            searchable_text = f"{item.get('사건번호', '')} {item.get('판시사항', '')} {item.get('판결요지', '')}"
-            if query.lower() in searchable_text.lower():
-                case_info = f"📌 사건번호: {item.get('사건번호', '')}\n"
-                case_info += f"📝 판시사항: {item.get('판시사항', '')[:200]}...\n"
-                case_info += f"⚖️ 판결요지: {item.get('판결요지', '')[:200]}...\n"
-                relevant_info.append(case_info)
+            case_number = item.get('사건번호', '')
+            
+            # 사건번호로 검색하는 경우 정확히 일치하는 것만 반환
+            if is_case_number:
+                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 query.lower() in searchable_text.lower():
+                    case_info = (
+                        f"📌 사건번호: {case_number}\n"
+                        f"📅 선고일자: {item.get('선고일자', '')}\n"
+                        f"📝 판시사항:\n{item.get('판시사항', '')[:500]}...\n\n"
+                        f"⚖️ 판결요지:\n{item.get('판결요지', '')[:500]}..."
+                    )
+                    relevant_info.append(case_info)
             
-            if len(relevant_info) >= 3:  # 최대 3개까지만 수집
+            if len(relevant_info) >= 3 and not is_case_number:  # 일반 검색의 경우만 여러 개 반환
                 break
         
         if relevant_info:
-            return "\n\n관련 판례 정보:\n" + "\n---\n".join(relevant_info[:3])
-        return ""
+            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 "❌ 해당하는 판례를 찾을 수 없습니다."
+    
     except Exception as e:
         print(f"판례 검색 오류: {e}")
-        return ""
+        return f"판례 검색 중 오류가 발생했습니다: {str(e)}"
 
 
 SYSTEM_PREFIX = """저는 법률 전문 AI 어시스턴트 'GiniGEN Legal'입니다.