Mikkatsuki commited on
Commit
882e67f
·
verified ·
1 Parent(s): f6108ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -21
app.py CHANGED
@@ -1,21 +1,5 @@
1
- """
2
- 多策略 RAG 文件問答系統 v2 — ChromaDB + PDF/DOCX 版本(優化版)
3
-
4
- 安裝依賴:
5
- pip install gradio groq pypdf python-docx sentence-transformers numpy chromadb scikit-learn
6
-
7
- 執行:
8
- python multistrategy_rag_chromadb_docx_v2.py
9
- """
10
  !pip install chromadb
11
-
12
- from __future__ import annotations
13
-
14
- import os
15
- import re
16
- import time
17
- from pathlib import Path
18
- from typing import Any
19
 
20
  import chromadb
21
  import gradio as gr
@@ -119,7 +103,7 @@ class MultiStrategyRAG:
119
 
120
  return (
121
  f"✓ 成功載入 {self.source_name}\n"
122
- f"類型:{suffix.upper().lstrip('.')} · {stats}\n"
123
  f"{len(self.chunks)} 個主片段 · ChromaDB Session:{self.session_id}"
124
  )
125
  except Exception as exc:
@@ -468,7 +452,6 @@ HEADER_HTML = """
468
  </div>
469
  """
470
 
471
-
472
  def build_strategy_menu(selected: str = "semantic") -> str:
473
  cards = []
474
  for key, name, desc, icon in STRATEGY_INFO:
@@ -480,7 +463,7 @@ def build_strategy_menu(selected: str = "semantic") -> str:
480
  <div class="strat-desc">{desc}</div>
481
  </button>"""
482
  )
483
- return f'<div class="strat-grid">{"".join(cards)}</div>'
484
 
485
 
486
  STRATEGY_MENU_JS = """
@@ -626,4 +609,4 @@ def create_interface():
626
 
627
  if __name__ == "__main__":
628
  demo = create_interface()
629
- demo.launch(share=False, server_name="0.0.0.0")
 
 
 
 
 
 
 
 
 
 
1
  !pip install chromadb
2
+ !pip install python-docx groq pypdf sentence-transformers chromadb
 
 
 
 
 
 
 
3
 
4
  import chromadb
5
  import gradio as gr
 
103
 
104
  return (
105
  f"✓ 成功載入 {self.source_name}\n"
106
+ f"類型:{suffix.upper().lstrip('.')}\n"
107
  f"{len(self.chunks)} 個主片段 · ChromaDB Session:{self.session_id}"
108
  )
109
  except Exception as exc:
 
452
  </div>
453
  """
454
 
 
455
  def build_strategy_menu(selected: str = "semantic") -> str:
456
  cards = []
457
  for key, name, desc, icon in STRATEGY_INFO:
 
463
  <div class="strat-desc">{desc}</div>
464
  </button>"""
465
  )
466
+ return f'<div class="strat-grid">{" ".join(cards)}</div>'
467
 
468
 
469
  STRATEGY_MENU_JS = """
 
609
 
610
  if __name__ == "__main__":
611
  demo = create_interface()
612
+ demo.launch(share=False, server_name="0.0.0.0")