audit_assistant / app.py
ppsingh's picture
check
6fd4eb8
raw
history blame
792 Bytes
import gradio as gr
import pandas as pd
import logging
import asyncio
import os
from uuid import uuid4
from datetime import datetime
from pathlib import Path
from huggingface_hub import CommitScheduler
from auditqa.sample_questions import QUESTIONS
from auditqa.reports import files, report_list, new_files, new_report_list
from auditqa.process_chunks import load_chunks, getconfig, get_local_qdrant, load_new_chunks
from auditqa.retriever import get_context
from auditqa.reader import nvidia_client, dedicated_endpoint
from auditqa.utils import make_html_source, parse_output_llm_with_sources, save_logs, get_message_template
from dotenv import load_dotenv
load_dotenv()
vectorstores = get_local_qdrant()
vectorstore = vectorstores['allreports']
print(vectorstore)
print(type(vectorstore))