Update app.py

app.py

@@ -1,14 +1,11 @@
-import os
 import requests
-import zipfile
-import io
 import gradio as gr
 
 def get_file_summary(file_info):
     return {
-        "name": file_info.filename,
-        "type": "binary" if file_info.file_size > 1024 * 1024 else "text",
-        "size": file_info.file_size,
+        "name": file_info['path'],
+        "type": "binary" if file_info['size'] > 1024 * 1024 else "text",
+        "size": file_info['size'],
     }
 
 def extract_repo_content(url):
@@ -25,22 +22,22 @@ def extract_repo_content(url):
     
     repo_content = response.json()
     extracted_content = []
-    headers = []
     
     for file_info in repo_content:
         file_summary = get_file_summary(file_info)
-        headers.append(file_summary)
+        content = {"header": file_summary}
         
         if file_summary["type"] == "text" and file_summary["size"] <= 1024 * 1024:
-            file_url = f"https://huggingface.co/{repo_type}/{repo_name}/resolve/main/{file_info['filename']}"
+            file_url = f"https://huggingface.co/{repo_type}/{repo_name}/resolve/main/{file_info['path']}"
             file_response = requests.get(file_url)
             if file_response.status_code == 200:
-                file_content = file_response.text
-                extracted_content.append({"header": file_summary, "content": file_content})
+                content["content"] = file_response.text
             else:
-                extracted_content.append({"header": file_summary, "content": "Failed to fetch file content."})
+                content["content"] = "Failed to fetch file content."
         else:
-            extracted_content.append({"header": file_summary, "content": "File too large or binary, content not captured."})
+            content["content"] = "File too large or binary, content not captured."
+        
+        extracted_content.append(content)
     
     return extracted_content