lesgo2

app.py

@@ -46,11 +46,12 @@ def process_llm_response(llm_response):
         print(source.metadata['source']+ "Page Number: " + str(source.metadata['page']))
         response_data['sources'].append({"book": source.metadata['source'], "page": source.metadata['page']})
     # return json.dumps(response_data)
-    return llm_response['result']
+    return response_data
     
 def get_answer(question):
     llm_response = qa_chain(question)
     response = process_llm_response(llm_response)
+
     return response["result"], response["sources"]
 
 # @app.route('/question', methods=['POST'])
@@ -144,10 +145,17 @@ Use only text found in the context as your knowledge source for the answer.
 ----------------
 {context}"""
 
+
+def print_array(arr):
+    # Convert the array to a string representation
+    arr_str = str(arr)
+    return arr_str
+
 def getanswer(question):
     llm_response = qa_chain(question)
     response = process_llm_response(llm_response)
-    return response
+    sources= print_array(response["sources"])
+    return response["result"], sources
 
 # iface = gr.Interface(fn=getanswer, inputs="text", outputs="text")
 # iface.launch()