Update app.py

app.py

@@ -23,7 +23,6 @@ Answer the question based on the above context: {question}
 """
 
 def load_documents():
-
     loader = DirectoryLoader(DATA_PATH, glob="*.pdf")
     documents = loader.load()
     return documents
@@ -64,7 +63,12 @@ def get_response(query_text):
     if len(results) == 0 or results[0][1] < 0.7:
         print(f"Unable to find matching results.")
         return
-     
+
+    sources = []
+    for doc, score in results:
+        source_with_page = f"{doc.metadata.get('source', None)} (Page {doc.page_number})"
+        sources.append(source_with_page)
+
     context_text = "\n\n---\n\n".join([doc.page_content for doc, _score in results])
  	  
     context_text = "\n\n---\n\n".join([doc.page_content for doc, _score in results])
@@ -74,25 +78,25 @@ def get_response(query_text):
     model = ChatOpenAI()
     response_text = model.predict(prompt)
 	  
-    sources = [doc.metadata.get("source", None) for doc, _score in results]
-    formatted_response = f"Response: {response_text}\nSources: {sources}"
+    #sources = [doc.metadata.get("source", None) for doc, _score in results]
+    
+    formatted_response = f"Response: {response_text}\nSources: {', '.join(sources)}"
     return formatted_response
 	  
-def chatbot(query_text):
+def prepare():
     documents = load_documents()
     chunks = split_text(documents)
     save_to_chroma(chunks)
-    response = get_response(query_text)
-    return response
    
    
    	  
 
-iface = gr.Interface(fn=chatbot,
+iface = gr.Interface(fn=get_response,
         inputs=gr.components.Textbox(lines=7, label="Enter your text"),
         outputs="text",
         title="UK Insurance Law AI Tool")
      
 
 
+prepare()
 iface.launch()