'
return links_html
def get_saved_urls_list():
"""Get list of ingested URLs for dropdown"""
urls = []
if os.path.exists(INGESTED_URLS_FILE):
with open(INGESTED_URLS_FILE, "r") as f:
for line in f:
url = line.strip()
if url:
urls.append(url)
return urls
def delete_url_by_url(url_to_delete: str):
"""Delete URL by its actual URL string and its embeddings"""
if not os.path.exists(INGESTED_URLS_FILE):
return "No URLs to delete."
try:
with open(INGESTED_URLS_FILE, "r") as f:
urls = f.readlines()
# Find and remove the URL
found = False
for i, url in enumerate(urls):
if url.strip() == url_to_delete:
urls.pop(i)
found = True
break
if found:
with open(INGESTED_URLS_FILE, "w") as f:
f.writelines(urls)
# Delete embeddings for this URL
embeddings_result = delete_embeddings_by_source(url_to_delete)
return f"Deleted URL: {url_to_delete}\n{embeddings_result}"
else:
return f"URL not found: {url_to_delete}"
except Exception as e:
return f"Error deleting URL: {str(e)}"
def delete_uploaded_file(filename: str):
"""Delete an uploaded file and its embeddings"""
try:
file_path = os.path.join(UPLOAD_DIR, filename)
if os.path.exists(file_path):
# Delete the file
os.remove(file_path)
# Delete embeddings for this file
embeddings_result = delete_embeddings_by_source(file_path)
return f"Deleted file: {filename}\n{embeddings_result}"
else:
return f"File not found: {filename}"
except Exception as e:
return f"Error deleting file: {str(e)}"
def get_uploaded_files_list():
"""Get list of uploaded files with delete buttons"""
files = []
if os.path.exists(UPLOAD_DIR):
for filename in os.listdir(UPLOAD_DIR):
full_path = os.path.join(UPLOAD_DIR, filename)
if os.path.isfile(full_path):
files.append(filename)
return files
with gr.Blocks() as demo:
gr.Markdown("# 📘 Developer Docs Assistant")
with gr.Tab("Upload Document"):
with gr.Row():
with gr.Column(scale=2):
file = gr.File(
label="Upload Document",
file_types=[".pdf", ".txt", ".md", ".markdown"],
)
upload_btn = gr.Button("📤 Ingest File", variant="primary")
upload_output = gr.Textbox(label="Upload Result", visible=False)
# Progress indicator
upload_progress = gr.HTML(
value="
Ready to upload
",
label="Status",
)
with gr.Column(scale=1):
gr.Markdown("### 📋 Upload Instructions")
gr.Markdown(
"""
1. **Select File**: Choose a PDF, TXT, or Markdown file
2. **Click Upload**: The file will be processed and embedded
3. **Wait**: Processing may take a few moments
4. **Check Status**: Monitor the progress indicator
"""
)
def handle_upload_with_progress(file):
if not file:
return (
"⚠️ Please select a file first.",
gr.update(visible=True),
gr.update(
value="
❌ No file selected
"
),
)
# Show processing status
progress_html = """
🔄 Processing file...
"""
try:
result = handle_file_upload_with_progress(file)
success_html = f"""
",
label="Status",
)
with gr.Column(scale=1):
gr.Markdown("### 📋 URL Ingestion Instructions")
gr.Markdown(
"""
1. **Enter URL**: Paste a valid document URL
2. **Click Ingest**: Content will be fetched and processed
3. **Wait**: Processing may take a few moments
4. **Check Status**: Monitor the progress indicator
"""
)
def handle_url_ingestion_with_progress_ui(url):
if not url or not url.strip():
return (
"⚠️ Please enter a valid URL.",
gr.update(visible=True),
gr.update(
value="
❌ No URL provided
"
),
)
# Show processing status
progress_html = """
🔄 Fetching and processing URL...
"""
try:
result = handle_url_ingestion_with_progress(url.strip())
success_html = f"""