Enable all medical imaging tools (VQA, classification, report generation, DICOM, web browsing)

app.py

@@ -33,6 +33,7 @@ print(f"Using device: {device}")
 tools = []
 
 if device == "cuda":
+    # Load GPU-based tools
     try:
         from medrax.tools import XRayPhraseGroundingTool
         grounding_tool = XRayPhraseGroundingTool(
@@ -45,6 +46,59 @@ if device == "cuda":
     except Exception as e:
         print(f"✗ Failed to load grounding tool: {e}")
 
+    try:
+        from medrax.tools import XRayVQATool
+        vqa_tool = XRayVQATool(
+            device=device,
+            temp_dir="temp",
+            load_in_4bit=True
+        )
+        tools.append(vqa_tool)
+        print("✓ Loaded VQA tool")
+    except Exception as e:
+        print(f"✗ Failed to load VQA tool: {e}")
+
+    try:
+        from medrax.tools import XRayClassificationTool
+        classification_tool = XRayClassificationTool(
+            device=device,
+            temp_dir="temp",
+            load_in_4bit=True
+        )
+        tools.append(classification_tool)
+        print("✓ Loaded classification tool")
+    except Exception as e:
+        print(f"✗ Failed to load classification tool: {e}")
+
+    try:
+        from medrax.tools import XRayReportGenerationTool
+        report_tool = XRayReportGenerationTool(
+            device=device,
+            temp_dir="temp",
+            load_in_4bit=True
+        )
+        tools.append(report_tool)
+        print("✓ Loaded report generation tool")
+    except Exception as e:
+        print(f"✗ Failed to load report generation tool: {e}")
+
+# Load non-GPU tools
+try:
+    from medrax.tools import DICOMTool
+    dicom_tool = DICOMTool(temp_dir="temp")
+    tools.append(dicom_tool)
+    print("✓ Loaded DICOM tool")
+except Exception as e:
+    print(f"✗ Failed to load DICOM tool: {e}")
+
+try:
+    from medrax.tools import WebBrowsingTool
+    browsing_tool = WebBrowsingTool()
+    tools.append(browsing_tool)
+    print("✓ Loaded web browsing tool")
+except Exception as e:
+    print(f"✗ Failed to load web browsing tool: {e}")
+
 checkpointer = MemorySaver()
 
 llm = ModelFactory.create_model(