Update app.py

app.py

@@ -1,35 +1,28 @@
 import gradio as gr
 from PIL import Image
-from PIL.PngImagePlugin import PngInfo
 import requests
 from bs4 import BeautifulSoup
 import os
 import subprocess
-from huggingface_hub import ModelCard, SpaceCard
+from huggingface_hub import ModelCard, SpaceCard, HfApi
 
 # PNG Info Tool
 def png_info(file):
     img = Image.open(file)
-    info = img.info
-    return str(info)
+    metadata = img.info
+    metadata_str = "\n".join([f"{key}: {value}" for key, value in metadata.items()])
+    return metadata_str
 
-# Web Info Tool (with HTML Info)
+# Web Info Tool
 def web_info(url):
-    # Fetch the webpage
     response = requests.get(url)
     soup = BeautifulSoup(response.text, 'html.parser')
-    
-    # Extract basic info
-    title = soup.title.string if soup.title else "No Title"
+    title = soup.title.string if soup.title else ""
     meta_description = soup.find("meta", attrs={"name": "description"})
-    description = meta_description["content"] if meta_description else "No Description"
-    
-    # Extract additional HTML info
+    description = meta_description["content"] if meta_description else ""
     meta_tags = soup.find_all("meta")
     headers = {f"h{i}": len(soup.find_all(f"h{i}")) for i in range(1, 7)}
     links = [a["href"] for a in soup.find_all("a", href=True)]
-    
-    # Format the output
     basic_info = f"Title: {title}\nDescription: {description}"
     html_info = f"Meta Tags: {len(meta_tags)}\nHeaders: {headers}\nLinks: {len(links)}"
     return f"{basic_info}\n\nHTML Info:\n{html_info}"
@@ -45,37 +38,59 @@ def view_source_info(url):
 
 # Document Info Tool
 def document_info(file):
-    file_size = os.path.getsize(file)
-    file_type = os.path.splitext(file)[1]
-    return f"File Type: {file_type}\nFile Size: {file_size} bytes"
+    file_extension = os.path.splitext(file.name)[1].lower()
+    content = ""
+    preview = None
+    if file_extension in [".txt", ".md"]:
+        with open(file.name, "r") as f:
+            content = f.read()
+    elif file_extension in [".png", ".jpg", ".jpeg", ".bmp", ".gif"]:
+        preview = file.name
+    else:
+        content = "Unsupported file format"
+    return content, preview
 
 # Video Info Tool
 def video_info(file):
-    # Display the video
     video_display = file
-    
-    # Extract video metadata using ffprobe
     result = subprocess.run(
         ["ffprobe", "-v", "error", "-show_format", "-show_streams", file],
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,
         text=True,
     )
-    
-    # Return the video display and metadata
     return video_display, result.stdout
 
 # Model Info Tool
 def model_info(model_id):
-    # Fetch model card from Hugging Face Hub
-    card = ModelCard.load(model_id)
-    return card.content  # Return the model card content (markdown)
+    try:
+        card = ModelCard.load(model_id)
+        api = HfApi()
+        model_info = api.model_info(model_id)
+        by = model_info.author if model_info.author else ""
+        description = card.content if card.content else ""
+        related_spaces = model_info.siblings
+        related_spaces = [s.rfilename for s in related_spaces if s.rfilename.endswith(".py")]
+        related_spaces = "\n".join(related_spaces) if related_spaces else ""
+        install_instructions = f"pip install transformers\n\n# To use the model:\nfrom transformers import AutoModel, AutoTokenizer\nmodel = AutoModel.from_pretrained('{model_id}')\ntokenizer = AutoTokenizer.from_pretrained('{model_id}')"
+        return by, description, related_spaces, install_instructions
+    except Exception:
+        return "", "", "", ""
 
 # Space Info Tool
 def space_info(space_id):
-    # Fetch space card from Hugging Face Hub
-    card = SpaceCard.load(space_id)
-    return card.content  # Return the space card content (markdown)
+    try:
+        card = SpaceCard.load(space_id)
+        api = HfApi()
+        space_info = api.space_info(space_id)
+        makes = space_info.author if space_info.author else ""
+        description = card.content if card.content else ""
+        code_files = space_info.siblings
+        code_files = [s.rfilename for s in code_files if s.rfilename.endswith(".py")]
+        code = "\n".join(code_files) if code_files else ""
+        return makes, code, description
+    except Exception:
+        return "", "", ""
 
 # Gradio Interface
 with gr.Blocks() as demo:
@@ -83,7 +98,7 @@ with gr.Blocks() as demo:
     
     with gr.Tab("PNG Info"):
         png_input = gr.File(label="Upload PNG File")
-        png_output = gr.Textbox(label="PNG Info")
+        png_output = gr.Textbox(label="PNG Metadata", lines=10)
         png_button = gr.Button("Extract PNG Info")
         png_button.click(png_info, inputs=png_input, outputs=png_output)
     
@@ -101,9 +116,23 @@ with gr.Blocks() as demo:
     
     with gr.Tab("Document Info"):
         doc_input = gr.File(label="Upload Document")
-        doc_output = gr.Textbox(label="Document Info")
+        content_output = gr.Textbox(label="Content", lines=10, visible=False)
+        preview_output = gr.Image(label="Preview (Image)", visible=False)
         doc_button = gr.Button("Extract Document Info")
-        doc_button.click(document_info, inputs=doc_input, outputs=doc_output)
+        
+        def update_output_components(file):
+            if file is None:
+                return gr.Textbox(visible=False), gr.Image(visible=False)
+            file_extension = os.path.splitext(file.name)[1].lower()
+            if file_extension in [".txt", ".md"]:
+                return gr.Textbox(visible=True), gr.Image(visible=False)
+            elif file_extension in [".png", ".jpg", ".jpeg", ".bmp", ".gif"]:
+                return gr.Textbox(visible=False), gr.Image(visible=True)
+            else:
+                return gr.Textbox(visible=True), gr.Image(visible=False)
+        
+        doc_button.click(document_info, inputs=doc_input, outputs=[content_output, preview_output])
+        doc_input.change(update_output_components, inputs=doc_input, outputs=[content_output, preview_output])
     
     with gr.Tab("Video Info"):
         video_input = gr.Video(label="Upload Video")
@@ -114,15 +143,20 @@ with gr.Blocks() as demo:
     
     with gr.Tab("Model Info"):
         model_input = gr.Textbox(label="Enter Model ID", placeholder="e.g., bert-base-uncased")
-        model_output = gr.Markdown(label="Model Info")
+        by_output = gr.Textbox(label="By", value="")
+        description_output = gr.Textbox(label="Description", value="", lines=5)
+        spaces_output = gr.Textbox(label="Related Spaces", value="", lines=3)
+        install_output = gr.Textbox(label="How to Install", value="", lines=5)
         model_button = gr.Button("Fetch Model Info")
-        model_button.click(model_info, inputs=model_input, outputs=model_output)
+        model_button.click(model_info, inputs=model_input, outputs=[by_output, description_output, spaces_output, install_output])
     
     with gr.Tab("Space Info"):
         space_input = gr.Textbox(label="Enter Space ID", placeholder="e.g., gradio/hello-world")
-        space_output = gr.Markdown(label="Space Info")
+        makes_output = gr.Textbox(label="Makes", value="")
+        code_output = gr.Textbox(label="Code", value="", lines=5)
+        description_output = gr.Textbox(label="Description", value="", lines=10)
         space_button = gr.Button("Fetch Space Info")
-        space_button.click(space_info, inputs=space_input, outputs=space_output)
+        space_button.click(space_info, inputs=space_input, outputs=[makes_output, code_output, description_output])
 
 # Launch the Gradio app
 demo.launch()