Update app.py

Improvements Made:
Error Handling:
Added handling for invalid URLs and failed API responses to return structured error messages.
Data Structure Integrity:
Ensured that extracted_content always contains dictionaries with a proper structure by initializing error messages as dictionaries.
Debugging Support:
Added checks in format_output to handle unexpected data formats gracefully and indicate potential data structure issues.Improvements Made:
Error Handling:
Added handling for invalid URLs and failed API responses to return structured error messages.
Data Structure Integrity:
Ensured that extracted_content always contains dictionaries with a proper structure by initializing error messages as dictionaries.
Debugging Support:
Added checks in format_output to handle unexpected data formats gracefully and indicate potential data structure issues.

app.py

@@ -1,5 +1,6 @@
 import requests
 import gradio as gr
+import json
 
 def get_file_summary(file_info):
     return {
@@ -10,7 +11,7 @@ def get_file_summary(file_info):
 
 def extract_repo_content(url):
     if "huggingface.co" not in url:
-        return "Invalid URL. Please provide a valid Hugging Face URL."
+        return [{"header": {"name": "Error", "type": "error", "size": 0}, "content": "Invalid URL. Please provide a valid Hugging Face URL."}]
     
     repo_name = url.split('/')[-2]
     repo_type = url.split('/')[-3]
@@ -18,7 +19,7 @@ def extract_repo_content(url):
     
     response = requests.get(api_url)
     if response.status_code != 200:
-        return f"Failed to fetch repository content. Status code: {response.status_code}"
+        return [{"header": {"name": "Error", "type": "error", "size": 0}, "content": f"Failed to fetch repository content. Status code: {response.status_code}"}]
     
     repo_content = response.json()
     extracted_content = []
@@ -44,11 +45,14 @@ def extract_repo_content(url):
 def format_output(extracted_content):
     formatted_output = ""
     for file_data in extracted_content:
-        formatted_output += f"### File: {file_data['header']['name']}\n"
-        formatted_output += f"**Type:** {file_data['header']['type']}\n"
-        formatted_output += f"**Size:** {file_data['header']['size']} bytes\n"
-        formatted_output += "#### Content:\n"
-        formatted_output += f"```\n{file_data['content']}\n```\n\n"
+        if isinstance(file_data, dict) and 'header' in file_data:
+            formatted_output += f"### File: {file_data['header']['name']}\n"
+            formatted_output += f"**Type:** {file_data['header']['type']}\n"
+            formatted_output += f"**Size:** {file_data['header']['size']} bytes\n"
+            formatted_output += "#### Content:\n"
+            formatted_output += f"```\n{file_data['content']}\n```\n\n"
+        else:
+            formatted_output += "Error in file data format.\n"
     return formatted_output
 
 def extract_and_display(url):