Update app.py

app.py

@@ -1,15 +1,13 @@
 import gradio as gr
 from huggingface_hub import list_models
-from reportlab.lib.pagesizes import letter
-from reportlab.pdfgen import canvas
+from openpyxl import Workbook
 import tempfile
-import os
 
 def get_models(username):
     if not username.strip():
-        result = [["-", "Please enter a valid username or organization.", "-", "-"]]
-        pdf_path = create_pdf(result)
-        return result, pdf_path
+        result = [["-", "Please enter a valid username or organization.", "-", "-", "-"]]
+        xlsx_path = create_excel(result)
+        return [row[:-1] for row in result], xlsx_path
 
     try:
         models = list_models(author=username.strip())
@@ -33,70 +31,45 @@ def get_models(username):
                 model_url
             ])
         if results:
-            pdf_path = create_pdf(results)
-            table_results = [row[:4] for row in results]
-            return table_results, pdf_path
+            xlsx_path = create_excel(results)
+            table_results = [row[:-1] for row in results]  # Do not show URL column in Gradio table
+            return table_results, xlsx_path
         else:
-            result = [["-", f"No models found for '{username}'.", "-", "-"]]
-            pdf_path = create_pdf(result)
-            return result, pdf_path
+            result = [["-", f"No models found for '{username}'.", "-", "-", "-"]]
+            xlsx_path = create_excel(result)
+            return [row[:-1] for row in result], xlsx_path
     except Exception as e:
-        result = [["-", f"Error: {str(e)}", "-", "-"]]
-        pdf_path = create_pdf(result)
-        return result, pdf_path
+        result = [["-", f"Error: {str(e)}", "-", "-", "-"]]
+        xlsx_path = create_excel(result)
+        return [row[:-1] for row in result], xlsx_path
 
-def create_pdf(data_rows):
-    # Create a temporary file for PDF
-    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-    p = canvas.Canvas(tmp.name, pagesize=letter)
-    width, height = letter
-    x_margin = 50
-    y_margin = 30
-    max_rows_per_page = 40
-    line_height = 20
-    p.setFont("Helvetica-Bold", 16)
-    p.drawString(x_margin, height - y_margin, "Hugging Face Model Explorer Results")
-    p.setFont("Helvetica", 11)
-    y = height - y_margin - 30
+def create_excel(data_rows):
     headers = ["Model Name", "Description", "Downloads", "Tags", "URL"]
-    p.setFont("Helvetica-Bold", 12)
-    p.drawString(x_margin, y, " | ".join(headers))
-    y -= line_height
-    p.setFont("Helvetica", 10)
-    row_count = 0
+    wb = Workbook()
+    ws = wb.active
+    ws.append(headers)
     for row in data_rows:
-        printable = [str(cell).replace('\n', ' ') for cell in row]
-        if len(printable) < 5:
-            printable += [""] * (5 - len(printable))
-        p.drawString(x_margin, y, " | ".join(printable))
-        y -= line_height
-        row_count += 1
-        if row_count % max_rows_per_page == 0:
-            p.showPage()
-            y = height - y_margin - 30
-            p.setFont("Helvetica-Bold", 12)
-            p.drawString(x_margin, y, " | ".join(headers))
-            y -= line_height
-            p.setFont("Helvetica", 10)
-    p.save()
+        ws.append(row)
+    tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".xlsx")
+    wb.save(tmp.name)
     tmp.close()
-    return tmp.name  # Return path for gr.File or gr.Download
+    return tmp.name
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤗 Hugging Face Model Explorer\nEnter a username/org to see all public models and download info as PDF!")
+    gr.Markdown("## 🤗 Hugging Face Model Explorer\nEnter a username/org to see all public models and download info as Excel!")
     username = gr.Textbox(label="HuggingFace Username or Organization", placeholder="e.g. Snaseem2026")
     table = gr.Dataframe(headers=["Name", "Description", "Downloads", "Tags"], interactive=False)
-    download_pdf = gr.File(label="Download PDF")  # gr.File for static files
+    download_excel = gr.File(label="Download Excel")  # Shows a download link/button
 
-    def search_and_pdf(user):
-        df, pdf_path = get_models(user)
-        return df, pdf_path
+    def search_and_excel(user):
+        df, excel_path = get_models(user)
+        return df, excel_path
 
-    search_btn = gr.Button("Search & Generate PDF")
+    search_btn = gr.Button("Search & Generate Excel")
     search_btn.click(
-        search_and_pdf,
+        search_and_excel,
         inputs=username,
-        outputs=[table, download_pdf]
+        outputs=[table, download_excel]
     )
 
 if __name__ == "__main__":