Ali2206 commited on
Commit
15023f4
·
verified ·
1 Parent(s): 41346ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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 = tempfile.mkdtemp(prefix="reports_") # fix for Gradio file serving
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{{kw}}\\b', page_text.lower()) for kw in MEDICAL_KEYWORDS):
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
- share=True
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
  )