Update app.py

app.py

@@ -92,7 +92,7 @@ class BasicAgent:
     
         if file_info.endswith((".png", ".jpg", ".jpeg")):
             print("coso Image file detected, processing with GPT-4o")
-            image = self._load_image(file_info)
+            image = get_or_download_image(file_info)
             response = self._ask_gpt4o_with_image(image, text)
             return response
 
@@ -157,7 +157,52 @@ class BasicAgent:
 
 
 
+def get_or_download_image(file_name: str) -> Image.Image:
+    """
+    Controlla se l'immagine è presente in /data. Se non lo è, la scarica da Hugging Face.
+    Restituisce un oggetto PIL.Image.
+    """
+    file_path = os.path.join("data", file_name)
+
+    if not os.path.exists(file_path):
+        print(f"[INFO] File {file_name} non trovato in /data, lo scarico...")
+        url = f"https://huggingface.co/datasets/gaia-benchmark/GAIA/resolve/main/2023/validation/{file_name}"
+        try:
+            response = requests.get(url)
+            response.raise_for_status()
+            with open(file_path, "wb") as f:
+                f.write(response.content)
+            print(f"[INFO] Scaricato e salvato in {file_path}")
+        except Exception as e:
+            print(f"[ERRORE] Impossibile scaricare l'immagine: {e}")
+            return None
 
+    try:
+        return Image.open(file_path)
+    except Exception as e:
+        print(f"[ERRORE] Impossibile aprire l'immagine {file_path}: {e}")
+        return None
+
+def download_file_if_needed(file_name):
+    if not file_name:
+        return None
+
+    local_path = f"/mnt/data/{file_name}"
+    if os.path.exists(local_path):
+        return local_path
+
+    base_url = "https://huggingface.co/datasets/gaia-benchmark/GAIA/resolve"
+    commit_hash = "86620fe7a265fdd074ea8d8c8b7a556a1058b0af"
+    full_url = f"{base_url}/{commit_hash}/2023/validation/{file_name}"
+
+    response = requests.get(full_url)
+    if response.status_code == 200:
+        with open(local_path, "wb") as f:
+            f.write(response.content)
+        return local_path
+    else:
+        print(f"Failed to download {file_name}: {response.status_code}")
+        return None
 
 
 def create_mock_questions():