Update app.py

app.py

@@ -147,14 +147,19 @@ def retrieve_document(query):
 
     print(filename)
 
-    pdf_file = requests.get(API_URL_FILES, params={"file":filename})
+    response = requests.get(API_URL_FILES, params={"file":filename})
 
-    print(pdf_file)
+    print(response)
 
-    # Read and return the document content
-    with open(pdf_file, "r", encoding="utf-8") as f:
-        return f.read()
+    if response.status_code == 200:
+        with open(filename, "wb") as f:
+            f.write(response.content)
+        print(f"File downloaded successfully: {filename}")
+    else:
+        print(f"Error: {response.status_code}, {response.json()}")
 
+    return filename
+    
 def clean_text(text):
     """Cleans extracted text for better processing by the model."""
     print("cleaning")