Add Hugging Face Space deployment files

- Add app.py with Gradio interface for HF Spaces
- Add requirements.txt with all dependencies
- Update README.md with HF Space configuration

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <[email protected]>

README.md

@@ -1,3 +1,15 @@
+---
+title: MedRAX2
+emoji: 🏥
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: 5.9.1
+app_file: app.py
+pinned: false
+license: mit
+---
+
 <h1 align="center">
 🤖 MedRAX-2: Medical Reasoning Agent for Chest X-ray
 </h1>

app.py

@@ -0,0 +1,94 @@
+"""
+MedRax2 Gradio Interface for Hugging Face Spaces
+Simple standalone version for deployment
+"""
+import os
+os.environ["OMP_NUM_THREADS"] = "1"
+
+import warnings
+warnings.filterwarnings("ignore")
+
+from huggingface_hub import login
+hf_token = os.environ.get("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+    print("✓ Logged in to HuggingFace")
+
+import gradio as gr
+from dotenv import load_dotenv
+import torch
+
+load_dotenv()
+
+from langgraph.checkpoint.memory import MemorySaver
+from medrax.models import ModelFactory
+from medrax.agent import Agent
+from medrax.utils import load_prompts_from_file
+
+os.makedirs("temp", exist_ok=True)
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
+tools = []
+
+if device == "cuda":
+    try:
+        from medrax.tools import XRayPhraseGroundingTool
+        grounding_tool = XRayPhraseGroundingTool(
+            device=device,
+            temp_dir="temp",
+            load_in_4bit=True
+        )
+        tools.append(grounding_tool)
+        print("✓ Loaded grounding tool")
+    except Exception as e:
+        print(f"✗ Failed to load grounding tool: {e}")
+
+checkpointer = MemorySaver()
+
+llm = ModelFactory.create_model(
+    model_name="gemini-2.0-flash",
+    temperature=0.7,
+    max_tokens=5000
+)
+
+prompts = load_prompts_from_file("medrax/docs/system_prompts.txt")
+prompt = prompts.get("MEDICAL_ASSISTANT", "You are a helpful medical imaging assistant.")
+
+agent = Agent(
+    llm,
+    tools=tools,
+    system_prompt=prompt,
+    checkpointer=checkpointer,
+)
+
+print(f"Tools loaded: {len(tools)}")
+
+def chat(message, history):
+    config = {"configurable": {"thread_id": "default"}}
+
+    # Handle multimodal input
+    if isinstance(message, dict):
+        text = message.get("text", "")
+        files = message.get("files", [])
+        if files:
+            file_info = f"[Image uploaded: {files[0]}]\n\n"
+            text = file_info + text
+        message = text
+
+    response = agent.workflow.invoke(
+        {"messages": [("user", message)]},
+        config=config
+    )
+    return response["messages"][-1].content
+
+demo = gr.ChatInterface(
+    fn=chat,
+    title="MedRAX2 - Medical AI Assistant",
+    description=f"Device: {device} | Tools: {len(tools)} loaded",
+    multimodal=True
+)
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -0,0 +1,296 @@
+accelerate==1.12.0
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.13.2
+aiohttp-retry==2.9.1
+aiosignal==1.4.0
+albucore==0.0.24
+albumentations==2.0.8
+annotated-doc==0.0.4
+annotated-types==0.7.0
+anthropic==0.75.0
+antlr4-python3-runtime==4.9.3
+anyio==4.12.0
+appnope==0.1.4
+argon2-cffi==25.1.0
+argon2-cffi-bindings==25.1.0
+arrow==1.4.0
+asttokens==3.0.1
+async-lru==2.0.5
+attrs==25.4.0
+babel==2.17.0
+backoff==2.2.1
+bcrypt==5.0.0
+beautifulsoup4==4.14.3
+bitsandbytes==0.49.0
+bleach==6.3.0
+brotli==1.2.0
+build==1.3.0
+cachetools==6.2.4
+certifi==2025.11.12
+cffi==2.0.0
+charset-normalizer==3.4.4
+chromadb==1.3.7
+click==8.2.1
+cohere==5.20.0
+coloredlogs==15.0.1
+comm==0.2.3
+contourpy==1.3.3
+cryptography==46.0.3
+cycler==0.12.1
+dataclasses-json==0.6.7
+datasets==4.4.1
+debugpy==1.8.19
+decorator==5.2.1
+defusedxml==0.7.1
+diffusers==0.36.0
+dill==0.4.0
+distro==1.9.0
+docstring_parser==0.17.0
+duckduckgo_search==8.1.1
+durationpy==0.10
+einops==0.8.1
+einops-exts==0.0.4
+executing==2.2.1
+fastapi==0.124.4
+fastavro==1.12.1
+fastjsonschema==2.21.2
+ffmpy==1.0.0
+filelock==3.20.1
+filetype==1.2.0
+flatbuffers==25.9.23
+fonttools==4.61.1
+fqdn==1.5.1
+frozenlist==1.8.0
+fsspec==2025.10.0
+google-auth==2.45.0
+google-genai==1.55.0
+googleapis-common-protos==1.72.0
+gradio==6.1.0
+gradio_client==2.0.1
+groovy==0.1.2
+grpcio==1.76.0
+h11==0.16.0
+hf-xet==1.2.0
+httpcore==1.0.9
+httptools==0.7.1
+httpx==0.28.1
+httpx-sse==0.4.0
+huggingface-hub==0.36.0
+humanfriendly==10.0
+hydra-core==1.3.2
+idna==3.11
+ImageIO==2.37.2
+importlib_metadata==8.7.0
+importlib_resources==6.5.2
+iopath==0.1.10
+ipykernel==7.1.0
+ipython==9.8.0
+ipython_pygments_lexers==1.1.1
+ipywidgets==8.1.8
+isoduration==20.11.0
+jedi==0.19.2
+Jinja2==3.1.6
+jiter==0.12.0
+joblib==1.5.3
+json5==0.12.1
+jsonpatch==1.33
+jsonpointer==3.0.0
+jsonschema==4.25.1
+jsonschema-specifications==2025.9.1
+jupyter==1.1.1
+jupyter-console==6.6.3
+jupyter-events==0.12.0
+jupyter-lsp==2.3.0
+jupyter_client==8.7.0
+jupyter_core==5.9.1
+jupyter_server==2.17.0
+jupyter_server_terminals==0.5.3
+jupyterlab==4.5.1
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.28.0
+jupyterlab_widgets==3.0.16
+kiwisolver==1.4.9
+kubernetes==34.1.0
+langchain==1.2.0
+langchain-anthropic==1.3.0
+langchain-chroma==1.1.0
+langchain-classic==1.0.0
+langchain-cohere==0.5.0
+langchain-community==0.4.1
+langchain-core==1.2.2
+langchain-google-genai==4.1.1
+langchain-openai==1.1.4
+langchain-pinecone==0.2.13
+langchain-text-splitters==1.1.0
+langchain-xai==1.1.0
+langgraph==1.0.5
+langgraph-checkpoint==3.0.1
+langgraph-prebuilt==1.0.5
+langgraph-sdk==0.3.0
+langsmith==0.5.0
+lark==1.3.1
+latex2mathml==3.78.1
+lazy_loader==0.4
+lxml==6.0.2
+markdown-it-py==4.0.0
+markdown2==2.5.4
+MarkupSafe==3.0.3
+marshmallow==3.26.1
+matplotlib==3.10.8
+matplotlib-inline==0.2.1
+mdurl==0.1.2
+-e git+https://github.com/bowang-lab/MedRAX2.git@95ddfff0fe4bf8aa271c38302447d9b85181a897#egg=medrax
+mistune==3.1.4
+mmh3==5.2.0
+mpmath==1.3.0
+msgpack==1.1.2
+multidict==6.7.0
+multiprocess==0.70.18
+mypy_extensions==1.1.0
+nbclient==0.10.2
+nbconvert==7.16.6
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.6.1
+notebook==7.5.1
+notebook_shim==0.2.4
+numpy==2.2.6
+oauthlib==3.3.1
+omegaconf==2.3.0
+onnxruntime==1.23.2
+openai==2.13.0
+opencv-python==4.12.0.88
+opencv-python-headless==4.12.0.88
+opentelemetry-api==1.39.1
+opentelemetry-exporter-otlp-proto-common==1.39.1
+opentelemetry-exporter-otlp-proto-grpc==1.39.1
+opentelemetry-proto==1.39.1
+opentelemetry-sdk==1.39.1
+opentelemetry-semantic-conventions==0.60b1
+orjson==3.11.5
+ormsgpack==1.12.1
+overrides==7.7.0
+packaging==24.2
+pandas==2.3.3
+pandocfilters==1.5.1
+parso==0.8.5
+pdfminer.six==20251107
+pdfplumber==0.11.8
+peft==0.18.0
+pexpect==4.9.0
+pillow==12.0.0
+pinecone==7.3.0
+pinecone-client==6.0.0
+pinecone-plugin-assistant==1.8.0
+pinecone-plugin-interface==0.0.7
+platformdirs==4.5.1
+portalocker==3.2.0
+posthog==5.4.0
+primp==0.15.0
+prometheus_client==0.23.1
+prompt_toolkit==3.0.52
+propcache==0.4.1
+protobuf==6.33.2
+psutil==7.1.3
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pyarrow==22.0.0
+pyasn1==0.6.1
+pyasn1_modules==0.4.2
+pybase64==1.4.3
+pycparser==2.23
+pydantic==2.12.4
+pydantic-settings==2.12.0
+pydantic_core==2.41.5
+pydicom==3.0.1
+pydub==0.25.1
+Pygments==2.19.2
+pylibjpeg==2.1.0
+pyngrok==7.5.0
+pyparsing==3.2.5
+PyPDF2==3.0.1
+pypdfium2==5.2.0
+PyPika==0.48.9
+pyproject_hooks==1.2.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.2.1
+python-json-logger==4.0.0
+python-multipart==0.0.20
+pytz==2025.2
+PyYAML==6.0.3
+pyzmq==27.1.0
+ray==2.52.1
+referencing==0.37.0
+regex==2025.11.3
+requests==2.32.5
+requests-oauthlib==2.0.0
+requests-toolbelt==1.0.0
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rfc3987-syntax==1.1.0
+rich==14.2.0
+rpds-py==0.30.0
+rsa==4.9.1
+safehttpx==0.1.7
+safetensors==0.7.0
+scikit-image==0.25.2
+scikit-learn==1.8.0
+scipy==1.16.3
+seaborn==0.13.2
+semantic-version==2.10.0
+Send2Trash==1.8.3
+sentencepiece==0.2.1
+setuptools==80.9.0
+shellingham==1.5.4
+shortuuid==1.0.13
+simsimd==6.5.8
+six==1.17.0
+sniffio==1.3.1
+soupsieve==2.8
+SQLAlchemy==2.0.45
+stack-data==0.6.3
+starlette==0.50.0
+stringzilla==4.5.1
+svgwrite==1.4.3
+sympy==1.14.0
+tenacity==9.1.2
+terminado==0.18.1
+threadpoolctl==3.6.0
+tifffile==2025.12.12
+tiktoken==0.12.0
+timm==0.5.4
+tinycss2==1.4.0
+tokenizers==0.20.3
+tomlkit==0.13.3
+torch==2.9.1
+torchvision==0.24.1
+torchxrayvision==1.4.0
+tornado==6.5.4
+tqdm==4.67.1
+traitlets==5.14.3
+transformers @ git+https://github.com/huggingface/transformers.git@88d960937c81a32bfb63356a2e8ecf7999619681
+typer==0.20.0
+types-PyYAML==6.0.12.20250915
+types-requests==2.32.4.20250913
+typing-inspect==0.9.0
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+tzdata==2025.3
+uri-template==1.3.0
+urllib3==2.3.0
+uuid_utils==0.12.0
+uvicorn==0.38.0
+uvloop==0.22.1
+watchfiles==1.1.1
+wavedrom==2.0.3.post3
+wcwidth==0.2.14
+webcolors==25.10.0
+webencodings==0.5.1
+websocket-client==1.9.0
+websockets==15.0.1
+widgetsnbextension==4.0.15
+xxhash==3.6.0
+yarl==1.22.0
+zipp==3.23.0
+zstandard==0.25.0