Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from datetime import datetime
 
 #Fill in your correct configuration
-invoke_url = 'https://75r40aboqb.execute-api.us-west-2.amazonaws.com/prod'
+invoke_url = 'https://8pkry7xzo5.execute-api.us-west-2.amazonaws.com/prod'
 bedrock_url = 'https://8hsh7fxan7.execute-api.us-west-2.amazonaws.com/prod'
 
 
@@ -239,6 +239,8 @@ def get_answer(task_type,question,sessionId,language,modelType,prompt,searchEngi
     print("answer:",answer)
 
     source_str = ""
+    query_docs_score_list = []
+    answer_docs_score_list = []
     for i in range(len(source_list)):
         item = source_list[i]
         print('item:',item)
@@ -252,19 +254,22 @@ def get_answer(task_type,question,sessionId,language,modelType,prompt,searchEngi
         except KeyError:
             source ="source:unknown"
             print("KeyError:source file not found")
-        score = "score:" + str(item['score'])
+        qd_score = "qd score:" + str(item['scoreQueryDoc'])
+        query_docs_score_list.append(item['scoreQueryDoc'])
+
+        ad_score = "ad score:" + str(item['scoreAnswerDoc'])
+        answer_docs_score_list.append(item['scoreAnswerDoc'])
+
         sentence = "sentence:" + item['sentence']
         paragraph = "paragraph:" + item['paragraph']
-        source_str += (_id + "      " + source + "      " + score + '\n')
+
+        source_str += (_id + "      " + source + "      " + qd_score + '\n')
         # source_str += sentence + '\n'
         source_str += paragraph + '\n\n'
     
     confidence = ""
-    query_docs_score = -1
-    if 'scoreQueryDoc' in result.keys():
-        query_docs_score =  float(result['scoreQueryDoc'])
-    if query_docs_score >= 0:
-        confidence += ("query_docs_score:" + str(query_docs_score) + '\n')
+    if len(query_docs_score_list) >= 0:
+        confidence += ("query_docs_score:" + str(query_docs_score_list) + '\n')
 
     query_answer_score = -1
     if 'scoreQueryAnswer' in result.keys():
@@ -273,12 +278,8 @@ def get_answer(task_type,question,sessionId,language,modelType,prompt,searchEngi
         confidence += ("query_answer_score:" + str(query_answer_score) + '\n')
 
     answer_docs_score = -1
-    if 'scoreAnswerDoc' in result.keys():
-        answer_docs_score =  float(result['scoreAnswerDoc'])
-    if answer_docs_score >= 0:
-        confidence += ("answer_docs_score:" + str(answer_docs_score) + '\n')
-
-
+    if len(answer_docs_score_list) >= 0:
+        confidence += ("answer_docs_score:" + str(answer_docs_score_list) + '\n')
 
     return answer,confidence,source_str,url,request_time