Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import sys
 import traceback
 import os
-from huggingface_hub import login
+from huggingface_hub import login, list_repo_files
 
 def system_info():
     try:
@@ -27,6 +27,40 @@ def system_info():
     except Exception as e:
         return f"Error: {str(e)}\n\n{traceback.format_exc()}"
 
+def list_gemma3_files():
+    try:
+        result = []
+        result.append("Listing files in Gemma 3 repository...")
+        
+        # Get token from environment
+        token = os.environ.get("HUGGINGFACE_TOKEN", "")
+        if token:
+            result.append(f"Token found: {token[:5]}...")
+        else:
+            result.append("No token found in environment variables!")
+            return "\n".join(result)
+        
+        # Login to Hugging Face
+        try:
+            login(token=token)
+            result.append("Successfully logged in to Hugging Face Hub")
+        except Exception as e:
+            result.append(f"Error logging in: {e}")
+            return "\n".join(result)
+        
+        # List files in the repository
+        model_id = "google/gemma-3-27b-it-qat-q4_0-gguf"
+        result.append(f"Listing files in {model_id}...")
+        
+        files = list_repo_files(model_id, token=token)
+        result.append("Files found:")
+        for file in files:
+            result.append(f"- {file}")
+        
+        return "\n".join(result)
+    except Exception as e:
+        return f"Error: {str(e)}\n\n{traceback.format_exc()}"
+
 def test_gemma3():
     try:
         result = []
@@ -50,9 +84,22 @@ def test_gemma3():
         
         # Use Gemma 3 GGUF model
         model_id = "google/gemma-3-27b-it-qat-q4_0-gguf"
-        model_filename = "gemma-3-27b-it-qat-q4_0.gguf"
         
-        result.append(f"Downloading {model_id} if not already present...")
+        # First, list the files to find the correct filename
+        result.append(f"Listing files in {model_id} to find the model file...")
+        from huggingface_hub import list_repo_files
+        
+        files = list_repo_files(model_id, token=token)
+        gguf_files = [f for f in files if f.endswith('.gguf')]
+        
+        if not gguf_files:
+            result.append("No GGUF files found in the repository!")
+            return "\n".join(result)
+        
+        model_filename = gguf_files[0]  # Use the first GGUF file found
+        result.append(f"Found model file: {model_filename}")
+        
+        result.append(f"Downloading {model_id}/{model_filename}...")
         from huggingface_hub import hf_hub_download
         
         model_path = hf_hub_download(
@@ -118,6 +165,19 @@ with gr.Blocks(title="StaffManager AI Assistant") as demo:
             outputs=[info_result]
         )
     
+    with gr.Tab("List Gemma 3 Files"):
+        with gr.Row():
+            with gr.Column():
+                list_files_button = gr.Button("List Gemma 3 Files")
+            with gr.Column():
+                list_files_result = gr.Textbox(label="Files in Repository", lines=20)
+        
+        list_files_button.click(
+            fn=list_gemma3_files,
+            inputs=[],
+            outputs=[list_files_result]
+        )
+    
     with gr.Tab("Gemma 3 Test"):
         with gr.Row():
             with gr.Column():