Logging added
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ import streamlit as st
|
|
4 |
from pinecone import Pinecone, ServerlessSpec
|
5 |
from sentence_transformers import SentenceTransformer, util
|
6 |
from openai import OpenAI
|
|
|
|
|
7 |
import os
|
8 |
api_key='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'
|
9 |
|
@@ -21,6 +23,11 @@ pc_api_key= '3f916d01-2a69-457d-85eb-966c5d1849a8' #AWS
|
|
21 |
pc = Pinecone(api_key=pc_api_key)
|
22 |
index = pc.Index(name=INDEX_NAME)
|
23 |
|
|
|
|
|
|
|
|
|
|
|
24 |
def query_from_pinecone(index,namespace, question_embedding, top_k=3):
|
25 |
# get embedding from THE SAME embedder as the documents
|
26 |
|
@@ -92,6 +99,12 @@ if QUESTION:
|
|
92 |
|
93 |
st.write(url)
|
94 |
st.write(ans)
|
|
|
|
|
|
|
95 |
|
96 |
else:
|
97 |
st.write("No matches for query")
|
|
|
|
|
|
|
|
4 |
from pinecone import Pinecone, ServerlessSpec
|
5 |
from sentence_transformers import SentenceTransformer, util
|
6 |
from openai import OpenAI
|
7 |
+
from datetime import datetime
|
8 |
+
import pandas as pd
|
9 |
import os
|
10 |
api_key='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'
|
11 |
|
|
|
23 |
pc = Pinecone(api_key=pc_api_key)
|
24 |
index = pc.Index(name=INDEX_NAME)
|
25 |
|
26 |
+
try:
|
27 |
+
df_log=pd.read_csv('query.csv', index_col=0)
|
28 |
+
except:
|
29 |
+
df_log=pd.DataFrame(columns=['query','url', 'result', 'ts'])
|
30 |
+
|
31 |
def query_from_pinecone(index,namespace, question_embedding, top_k=3):
|
32 |
# get embedding from THE SAME embedder as the documents
|
33 |
|
|
|
99 |
|
100 |
st.write(url)
|
101 |
st.write(ans)
|
102 |
+
now= str(datetime.utcnow())
|
103 |
+
df_log.loc[len(df_log)]=[QUESTION,resp[0]['id'],ans,now]
|
104 |
+
df_log.to_csv('query.csv')
|
105 |
|
106 |
else:
|
107 |
st.write("No matches for query")
|
108 |
+
now= str(datetime.utcnow())
|
109 |
+
df_log.loc[len(df_log)]=[QUESTION,'No match','-',now]
|
110 |
+
df_log.to_csv('query.csv')
|