File size: 1,456 Bytes
d166583 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import gradio as gr
import os
import faiss
import numpy as np
from pathlib import Path
import shutil
# Constants
DATA_DIR = "data"
INDEX_PATH = os.path.join(DATA_DIR, "faiss_index.index")
def save_index(file_obj):
"""
Save uploaded FAISS index to the data directory
"""
# Create data directory if it doesn't exist
Path(DATA_DIR).mkdir(exist_ok=True)
# Check if index already exists
if os.path.exists(INDEX_PATH):
return "⚠️ A FAISS index already exists in the data directory. Please remove it first."
try:
# Copy the temporary file to our target location
shutil.copy2(file_obj.name, INDEX_PATH)
# Verify the saved file is a valid FAISS index
faiss.read_index(INDEX_PATH)
return "✅ FAISS index successfully uploaded and saved!"
except Exception as e:
# If there was an error, remove the file if it was created
if os.path.exists(INDEX_PATH):
os.remove(INDEX_PATH)
return f"❌ Error: Invalid FAISS index file - {str(e)}"
# Create Gradio interface
demo = gr.Interface(
fn=save_index,
inputs=gr.File(label="Upload FAISS Index", file_types=[".index"]),
outputs=gr.Textbox(label="Status"),
title="FAISS Index Uploader",
description="Upload a FAISS index file to store in the HuggingFace Space data directory.",
allow_flagging="never"
)
if __name__ == "__main__":
demo.launch() |