diff --git a/FindinDB.py b/FindinDB.py index 5602db0..d6cf1b6 100644 --- a/FindinDB.py +++ b/FindinDB.py @@ -86,8 +86,8 @@ def FinderDbs(query,dbs,filtred=False,th=1.2): AllData[output[0].metadata["id"]]["d"]=min([AllData[output[0].metadata["id"]]["d"]-0.1,output[1]-0.1]) else: AllData[output[0].metadata["id"]]={"d":output[1],"page_content":output[0].page_content} - for item in AllData.items(): - print(item) + #for item in AllData.items(): + # print(item) if filtred: filtredData={} @@ -105,6 +105,17 @@ def FinderDbs(query,dbs,filtred=False,th=1.2): def QARequest(Pregunta,filtred=False): query = Pregunta AllData=FinderDbs(query,[db,db2],filtred) + + if AllData: + import markdown + AllData = list(AllData) + lista = [] + for i in AllData[0].items(): + to_append = markdown.markdown(i[1]['page_content']) + lista.append(to_append) + lista.append('
') + + AllData[0] = lista return AllData diff --git a/requirements.txt b/requirements.txt index c412cef..efc33da 100644 --- a/requirements.txt +++ b/requirements.txt @@ -37,6 +37,7 @@ httptools==0.6.0 httpx==0.25.0 huggingface-hub==0.17.2 idna==3.4 +importlib-metadata==6.8.0 importlib-resources==6.1.0 Jinja2==3.1.2 joblib==1.3.2 @@ -47,6 +48,7 @@ langchain==0.0.297 langsmith==0.0.38 lit==16.0.6 lxml==4.9.3 +Markdown==3.4.4 MarkupSafe==2.1.3 marshmallow==3.20.1 matplotlib==3.7.3 diff --git a/run.sh b/run.sh index 829991f..b5ec8d4 100644 --- a/run.sh +++ b/run.sh @@ -1 +1 @@ -uvicorn FindinDB:app --reload \ No newline at end of file +uvicorn FindinDB:app --reload --port 7860 \ No newline at end of file