Update app.py

app.py

@@ -198,6 +198,30 @@ class HuggingFaceDocumentQATool:
         except Exception as e:
             return f"QA error: {str(e)}"
 
+
+
+from transformers import BlipProcessor, BlipForQuestionAnswering
+
+class HuggingFaceImageQATool(Tool):
+    name = "image_qa"
+    description = "Answer questions about an image."
+    inputs = {
+        "image_path": {"type": "string", "description": "Path to image"},
+        "question": {"type": "string", "description": "Question about the image"}
+    }
+    output_type = "string"
+
+    def __init__(self):
+        self.processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
+        self.model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
+
+    def forward(self, image_path: str, question: str) -> str:
+        image = Image.open(image_path)
+        inputs = self.processor(image, question, return_tensors="pt")
+        out = self.model.generate(**inputs)
+        return self.processor.decode(out[0], skip_special_tokens=True)
+
+
 # ===== Code Execution =====
 class PythonCodeExecutionTool:
     def __init__(self):