Update providers/archive_provider.py

providers/archive_provider.py

@@ -46,7 +46,6 @@ def search(query: str, has_transcript: bool = False):
         logging.error(f"An error occurred during search in archive_provider: {e}")
         return {"error": "Failed to perform search."}
 
-
 def get_details(identifier: str):
     """
     Retrieves key details for a single item from the Internet Archive, excluding the transcript.
@@ -56,20 +55,25 @@ def get_details(identifier: str):
         return item
 
     try:
-        # --- Find Thumbnail, Video, and Audio Files ---
+        # --- Find Thumbnail ---
         thumbnail_file = next((f for f in item.files if f.get('name') == '__ia_thumb.jpg'), None) or \
                          next((f for f in item.files if f.get('format') == 'Item Tile'), None) or \
                          next((f for f in item.files if f.get('format') == 'Thumbnail'), None)
         thumbnail_url = f"https://archive.org/download/{identifier}/{thumbnail_file['name']}" if thumbnail_file else None
 
-        video_file = next((f for f in item.files if 'MPEG4' in f.get('format', '')), None)
-        video_url = f"https://archive.org/download/{identifier}/{video_file['name']}" if video_file else None
-
+        # --- Find Video File ---
+        # 1. المحاولة الأولى: البحث بالامتداد .mp4
         video_file = next((f for f in item.files if f.get('name', '').lower().endswith('.mp4')), None)
-
-        # 2. المحاولة الثانية: إذا لم نجد، نبحث في خانة الـ Format (يحل مشكلة القوائم والأرشيف)
+        
+        # 2. المحاولة الثانية: البحث في الـ Format
         if not video_file:
             video_file = next((f for f in item.files if 'MPEG4' in f.get('format', '') or 'h.264' in f.get('format', '').lower()), None)
+        
+        video_url = f"https://archive.org/download/{identifier}/{video_file['name']}" if video_file else None
+
+        # --- Find Audio File ---
+        audio_file = next((f for f in item.files if f.get('format') == 'VBR MP3'), None)
+        audio_url = f"https://archive.org/download/{identifier}/{audio_file['name']}" if audio_file else None
 
         # --- Structure the final response ---
         item_metadata = item.metadata
@@ -89,3 +93,8 @@ def get_details(identifier: str):
     except Exception as e:
         logging.error(f"An error occurred getting details in archive_provider for {identifier}: {e}")
         return {"error": f"Failed to retrieve details for item {identifier}."}
+        return response
+
+    except Exception as e:
+        logging.error(f"An error occurred getting details in archive_provider for {identifier}: {e}")
+        return {"error": f"Failed to retrieve details for item {identifier}."}