Merge pull request #3 from Janarthanam/streamlit

README.md

@@ -1,5 +1,5 @@
 ---
-title: Toypdf
+title: retrieval_aug_llm
 emoji: 📉
 colorFrom: blue
 colorTo: red

api/routes/search.py

@@ -30,7 +30,8 @@ async def answer(name: str, query: str):
     docs = _db.similarity_search_with_score(query=query)
     print(docs)
     answer = _chain.run(input_documents=[tup[0] for tup in docs], question=query)
-    return JSONResponse(status_code=200, content={"answer": answer, "file_score": [[f"{d[0].metadata['file']} : {d[0].metadata['page']}", d[1]] for d in docs]}) 
+    return JSONResponse(status_code=200, content={"answer": answer, "metadata": [
+        {"file": d[0].metadata['file'], "page" : d[0].metadata['page'], "score": d[1]} for d in docs]}) 
 
 
 @router.get("/v1/datasets")

fe/app.py

@@ -16,10 +16,13 @@ if ds:
     query = st.text_input("Enter your search query",
                           placeholder="Ask your question")
     if query:
-        answer = requests.get(f"http://localhost:8080/v1/datasets/{ds}/answer?query={query}", 
-                       timeout=5000 )
+        res = requests.get(f"{BE}/v1/datasets/{ds}/answer?query={query}", 
+                       timeout=5000 ).json()
+        answer = res["answer"]
 
-        print(answer.json()["answer"])
-        st.write(answer.json()["answer"])
+        st.write(answer)
+        files = [f"{f['file']}, page {f['page']}" for f in res["metadata"]]
+        for fi in files:
+            st.markdown(f"- {fi}")
 else:
     st.write("Choose your dataset!")