Create app.py

app.py

@@ -0,0 +1,182 @@
+import json, gradio as gr, os, re
+from collections import deque
+
+# ==========================================
+# 1. 基础配置与数据加载 (严格执行约定)
+# ==========================================
+PROJECT_NAME = "家族亲戚关系网查询系统"
+DATA_FILE = "family_data.jsonl"
+PHOTO_ROOT = "photos"
+ARCHIVE_ROOT = "archives"
+
+def load_family_data():
+    db = {}
+    if not os.path.exists(DATA_FILE): return db
+    with open(DATA_FILE, "r", encoding="utf-8") as f:
+        for line in f:
+            try:
+                data = json.loads(line.strip())
+                if "n" in data: db[str(data["n"]).strip()] = data
+            except: continue
+    return db
+
+db = load_family_data()
+
+def get_year(name):
+    rec = db.get(name, {})
+    match = re.search(r'(\d{4})', str(rec.get("info", "")))
+    return int(match.group(1)) if match else 9999
+
+def fmt_member(n):
+    """格式化成员显示：姓名 + 生卒/在世 + 配偶"""
+    r = db.get(n, {})
+    if not r: return n
+    info = str(r.get("info", ""))
+    # 提取年份
+    years = ""
+    y_match = re.search(r'(\d{4}-\d{4}|\d{4}-)', info)
+    if y_match: years = f" ({y_match.group()})"
+    elif re.search(r'\d{4}', info): years = f" ({re.search(r'\d{4}', info).group()}-)"
+    
+    sps = r.get("sp", [])
+    sp_str = f" [配:{''.join(sps)}]" if sps and sps != ["无"] else ""
+    return f"{n}{sp_str}{years}"
+
+# ==========================================
+# 2. 核心算法：六大维度推演与自然语言解析
+# ==========================================
+
+def run_inference(start_name, cate):
+    start_name = start_name.strip()
+    me = db.get(start_name)
+    if not me: return f"❌ 未找到【{start_name}】的记录"
+    
+    res = f"📜 {start_name} 的【{cate}】系关系网\n" + "="*40 + "\n"
+    content = ""
+
+    # 获取父母与子女
+    my_f, my_m = me.get("f"), me.get("m")
+    children = [n for n, d in db.items() if d.get("f") == start_name or d.get("m") == start_name]
+
+    if cate == "晚辈":
+        def get_descendants(p, level=1):
+            kids = [n for n, d in db.items() if d.get("f") == p or d.get("m") == p]
+            out = ""
+            for k in sorted(kids, key=get_year):
+                out += f"{'  '*level} ┕ {fmt_member(k)}\n" + get_descendants(k, level + 1)
+            return out
+        content = get_descendants(start_name)
+
+    elif cate == "前辈":
+        def get_ancestors(p, level=1):
+            if level > 4 or not p or p == "待考": return ""
+            r = db.get(p, {})
+            line = f"{'  '*level} 第{level}代: {fmt_member(p)}\n"
+            return line + get_ancestors(r.get("f"), level+1) + get_ancestors(r.get("m"), level+1)
+        content = get_ancestors(my_f) + get_ancestors(my_m)
+
+    elif cate == "亲兄妹":
+        siblings = [n for n, d in db.items() if (d.get("f") == my_f and my_f != "待考") or (d.get("m") == my_m and my_m != "待考")]
+        for s in sorted(set(siblings), key=get_year):
+            if s != start_name: content += f" 🚩 {fmt_member(s)}\n"
+
+    elif cate == "姻兄妹":
+        for sp in me.get("sp", []):
+            sp_d = db.get(sp)
+            if sp_d:
+                s_f, s_m = sp_d.get("f"), sp_d.get("m")
+                sibs = [n for n, d in db.items() if (d.get("f") == s_f or d.get("m") == s_m) and n != sp]
+                content += f"\n💍 配偶【{sp}】的同胞：\n"
+                for s in sorted(set(sibs), key=get_year): content += f"   └── {fmt_member(s)}\n"
+
+    return res + (content if content else "（暂无数据记录）")
+
+# ==========================================
+# 3. 自然语言查询与档案展示
+# ==========================================
+
+def advanced_query(text):
+    """解析：张三的舅舅 / 张三,李四"""
+    text = text.replace("，", ",").strip()
+    if "的" in text:
+        name, rel = text.split("的")
+        p = db.get(name.strip())
+        if not p: return "找不到该人"
+        # 舅舅逻辑：母亲的兄弟
+        if "舅" in rel and p.get("m"):
+            m_d = db.get(p["m"])
+            if m_d and m_d.get("f"):
+                uncles = [n for n, d in db.items() if d.get("f") == m_d["f"] and d.get("s") == "男" and n != p["m"]]
+                return f"🔍 {name} 的舅舅有：\n" + "\n".join([fmt_member(u) for u in uncles])
+        return "暂未支持该特定称谓的自动解析"
+    
+    if "," in text:
+        n1, n2 = text.split(",")
+        # BFS 路径查找
+        queue = deque([(n1.strip(), [])])
+        visited = {n1.strip()}
+        while queue:
+            curr, path = queue.popleft()
+            if curr == n2.strip():
+                return "🔗 关系链：\n" + " ➔ ".join(path + [n2.strip()])
+            for neighbor, d in db.items():
+                if neighbor not in visited:
+                    # 检查父子或配偶关系
+                    if d.get("f") == curr or d.get("m") == curr or curr in d.get("sp", []):
+                        visited.add(neighbor)
+                        queue.append((neighbor, path + [curr]))
+        return "未发现关联路径"
+    return "格式错误，请输入：张三的舅舅 或 张三,李四"
+
+def get_personal_tab(name):
+    name = name.strip(); p = db.get(name)
+    if not p: return "❌ 找不到档案", None, gr.update(choices=[])
+    
+    info_md = f"### {name} 的个人资料\n- **性别**: {p.get('s','?')}\n- **生平**: {p.get('info','')}\n- **父母**: {p.get('f','待考')} & {p.get('m','待考')}\n- **配偶**: {'、'.join(p.get('sp', []))}"
+    
+    # 照片与过滤.gitkeep
+    photo = f"{PHOTO_ROOT}/{name}.jpg"
+    photo = photo if os.path.exists(photo) else None
+    files = []
+    if os.path.exists(f"{ARCHIVE_ROOT}/{name}"):
+        files = [f for f in os.listdir(f"{ARCHIVE_ROOT}/{name}") if not f.startswith(".")]
+    return info_md, photo, gr.update(choices=files)
+
+# ==========================================
+# 4. 统一 UI 界面
+# ==========================================
+with gr.Blocks(title=PROJECT_NAME, css="footer {display: none !important;}") as demo:
+    gr.Markdown(f"# 📜 {PROJECT_NAME}")
+    
+    with gr.Tabs():
+        with gr.Tab("📋 个人资料"):
+            name_in = gr.Textbox(label="输入姓名 (回车搜索)", placeholder="例如：冯宗德")
+            with gr.Row():
+                with gr.Column(scale=1):
+                    p_photo = gr.Image(label="人物像")
+                with gr.Column(scale=2):
+                    p_info = gr.Markdown("等待输入...")
+                    file_sel = gr.Dropdown(label="📁 档案资料库 (自动过滤.gitkeep)")
+            
+            # 档案内容展示区
+            out_txt = gr.Textbox(label="文字档案", lines=10, visible=False)
+            out_img = gr.Image(label="图片档案", visible=False)
+            
+            name_in.submit(get_personal_tab, inputs=name_in, outputs=[p_info, p_photo, file_sel])
+
+        with gr.Tab("🧬 族网推演"):
+            with gr.Row():
+                t_name = gr.Textbox(label="起点姓名", value="冯宗德")
+                t_cate = gr.Radio(["晚辈", "前辈", "亲兄妹", "姻兄妹"], label="维度", value="晚辈")
+            btn_run = gr.Button("开始推演", variant="primary")
+            t_out = gr.Textbox(label="推演结果", lines=20)
+            btn_run.click(run_inference, inputs=[t_name, t_cate], outputs=t_out)
+
+        with gr.Tab("🔍 关系溯源"):
+            q_in = gr.Textbox(label="输入查询内容", placeholder="例如：冯宗德的舅舅 或 冯仁杰,何宗涓")
+            q_btn = gr.Button("解析关系")
+            q_out = gr.Textbox(label="查询结果", lines=10)
+            q_btn.click(advanced_query, inputs=q_in, outputs=q_out)
+
+if __name__ == "__main__":
+    demo.launch()