Update app.py
Browse files
app.py
CHANGED
@@ -25,12 +25,13 @@ os.makedirs(base_dir, exist_ok=True)
|
|
25 |
model_cache_dir = os.path.join(base_dir, "txagent_models")
|
26 |
tool_cache_dir = os.path.join(base_dir, "tool_cache")
|
27 |
file_cache_dir = os.path.join(base_dir, "cache")
|
28 |
-
report_dir =
|
29 |
vllm_cache_dir = os.path.join(base_dir, "vllm_cache")
|
30 |
|
31 |
os.makedirs(model_cache_dir, exist_ok=True)
|
32 |
os.makedirs(tool_cache_dir, exist_ok=True)
|
33 |
os.makedirs(file_cache_dir, exist_ok=True)
|
|
|
34 |
os.makedirs(vllm_cache_dir, exist_ok=True)
|
35 |
|
36 |
os.environ.update({
|
@@ -63,7 +64,7 @@ def extract_priority_pages(file_path: str, max_pages: int = 20) -> str:
|
|
63 |
text_chunks.append(f"=== Page {i+1} ===\n{(page.extract_text() or '').strip()}")
|
64 |
for i, page in enumerate(pdf.pages[3:max_pages], start=4):
|
65 |
page_text = page.extract_text() or ""
|
66 |
-
if any(re.search(rf'\\b{
|
67 |
text_chunks.append(f"=== Page {i} ===\n{page_text.strip()}")
|
68 |
return "\n\n".join(text_chunks)
|
69 |
except Exception as e:
|
@@ -233,5 +234,5 @@ if __name__ == "__main__":
|
|
233 |
server_name="0.0.0.0",
|
234 |
server_port=7860,
|
235 |
show_error=True,
|
236 |
-
|
237 |
)
|
|
|
25 |
model_cache_dir = os.path.join(base_dir, "txagent_models")
|
26 |
tool_cache_dir = os.path.join(base_dir, "tool_cache")
|
27 |
file_cache_dir = os.path.join(base_dir, "cache")
|
28 |
+
report_dir = "/data/reports" # Persistent path for Hugging Face Spaces
|
29 |
vllm_cache_dir = os.path.join(base_dir, "vllm_cache")
|
30 |
|
31 |
os.makedirs(model_cache_dir, exist_ok=True)
|
32 |
os.makedirs(tool_cache_dir, exist_ok=True)
|
33 |
os.makedirs(file_cache_dir, exist_ok=True)
|
34 |
+
os.makedirs(report_dir, exist_ok=True)
|
35 |
os.makedirs(vllm_cache_dir, exist_ok=True)
|
36 |
|
37 |
os.environ.update({
|
|
|
64 |
text_chunks.append(f"=== Page {i+1} ===\n{(page.extract_text() or '').strip()}")
|
65 |
for i, page in enumerate(pdf.pages[3:max_pages], start=4):
|
66 |
page_text = page.extract_text() or ""
|
67 |
+
if any(re.search(rf'\\b{kw}\\b', page_text.lower()) for kw in MEDICAL_KEYWORDS):
|
68 |
text_chunks.append(f"=== Page {i} ===\n{page_text.strip()}")
|
69 |
return "\n\n".join(text_chunks)
|
70 |
except Exception as e:
|
|
|
234 |
server_name="0.0.0.0",
|
235 |
server_port=7860,
|
236 |
show_error=True,
|
237 |
+
allowed_paths=["/data/reports"]
|
238 |
)
|