Rename app.py to main.py

app.py → main.py

@@ -1,8 +1,31 @@
+import os
 import gradio as gr
 from repo_utils import extract_repo_content
-from display_utils import format_output
 
-# Extract and display function
+# Get the HF token and space author name from environment variables
+hf_token = os.getenv("HF_TOKEN")
+hf_user = os.getenv("SPACE_AUTHOR_NAME")
+
+if not hf_token:
+    raise ValueError("HF_TOKEN environment variable is not set")
+if not hf_user:
+    raise ValueError("SPACE_AUTHOR_NAME environment variable is not set")
+
+def format_output(extracted_content, repo_url):
+    formatted_output = f"# Repository URL: {repo_url}\n\n"
+    for file_data in extracted_content:
+        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 += f"**Created:** {file_data['header']['creation_date']}\n"
+            formatted_output += f"**Modified:** {file_data['header']['modification_date']}\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):
     extracted_content = extract_repo_content(url, hf_token, hf_user)
     formatted_output = format_output(extracted_content, url)