Update app.py

app.py

@@ -1,16 +1,6 @@
-import os
 import gradio as gr
 from repo_utils import extract_repo_content
 
-# 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:
@@ -27,6 +17,8 @@ def format_output(extracted_content, repo_url):
     return formatted_output
 
 def extract_and_display(url):
+    hf_token = os.getenv("HF_TOKEN")
+    hf_user = os.getenv("SPACE_AUTHOR_NAME")
     extracted_content = extract_repo_content(url, hf_token, hf_user)
     formatted_output = format_output(extracted_content, url)
     return formatted_output
@@ -34,10 +26,8 @@ def extract_and_display(url):
 app = gr.Blocks(theme="sudeepshouche/minimalist")
 
 with app:
-    gr.Markdown("# VectorSpace Explorer")
-    gr.Markdown("**Unleash the power of AI to explore Hugging Face repositories.**")
-    
-    url_input = gr.Textbox(label="🔗 Repository URL", placeholder="Enter the repository URL here OR select an example below...")
+    gr.Markdown("# Hugging Face Space / Model Repository Content Extractor")
+    url_input = gr.Textbox(label="https:// URL of Repository", placeholder="Enter the repository URL here OR select an example below...")
     url_examples = gr.Examples(
         examples=[
             ["https://huggingface.co/spaces/big-vision/paligemma-hf"],
@@ -47,9 +37,10 @@ with app:
         ],
         inputs=url_input
     )
-    output_display = gr.Textbox(label="📄 Extracted Repository Content", show_copy_button=True, lines=20, placeholder="Repository content will be extracted here...\n\nMetadata is captured for all files, but text content provided only for files less than 32 kb\n\n\n\nReview and search through the content here OR simply copy it for offline analysis!!. 🤖")
-    extract_button = gr.Button("🚀 Extract Content")
+    output_display = gr.Textbox(label="Extracted Repository Content", show_copy_button=True, lines=20, placeholder="Repository content will be extracted here...\n\nMetadata is captured for all files, but text content provided only for files less than 32 kb\n\n\n\nReview and search through the content here OR simply copy it for offline analysis!!. 🤖")
+    extract_button = gr.Button("Extract Content")
     
     extract_button.click(fn=extract_and_display, inputs=url_input, outputs=output_display)
 
-app.launch()
+if __name__ == "__main__":
+    app.launch()