sujitb commited on
Commit
c690f92
1 Parent(s): 8596e21

Logging added

Browse files
Files changed (1) hide show
  1. app.py +13 -0
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')