Update Index.py

Index.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 import os
 import json
 import pandas as pd
+import time
 
 import phoenix as px
 from phoenix.trace.langchain import OpenInferenceTracer, LangChainInstrumentor
@@ -193,10 +194,14 @@ def rag( question: str, evaluate: bool):
 
     #response = chain("how tredence brought good insight?")
     response = chain(_prompt(question))
-
     print(response["result"])
 
+    time.sleep(7)
+    
     df = px.active_session().get_spans_dataframe()
+    print(px.active_session())
+    print(px.active_session().get_spans_dataframe())
+    print(df.count())        
     df_sorted = df.sort_values(by='end_time',ascending=False) 
 
     model_input = json.loads(df_sorted[df_sorted["name"] == "LLMChain"]["attributes.input.value"][0])