Update app.py

app.py

@@ -23,9 +23,7 @@ pinecone.init(
 )
 
 index_name = PINECONE_INDEX
-start_time = time.time()
 vectorstore = Pinecone.from_existing_index(index_name=index_name, embedding=embeddings)
-print("---vectorstore %s seconds ---" % (time.time() - start_time))
 
 chain = get_chain(vectorstore)
 
@@ -51,8 +49,11 @@ class ChatWrapper:
             import openai
             openai.api_key = os.getenv("OPENAI_API_KEY")
             # Run chain and append input.
+            start_time = time.time()
             output = chain({"question": inp, "chat_history": history})["answer"]
+            print("---chain %s seconds ---" % (time.time() - start_time))
             history.append((inp, output))
+            print("---history %s seconds ---" % (time.time() - start_time))
         except Exception as e:
             raise e
         finally: