Update app.py

app.py

@@ -347,7 +347,7 @@ def rag_workflow(query):
 
     
     response = llm.invoke(prompt)
-    return response.content, references
+    return response.content
 
 
 def initialize():
@@ -395,25 +395,10 @@ import gradio as gr
 
 def bot_kadi(history):
     user_query = history[-1][0]
-    response, references = rag_workflow(user_query)
+    response = rag_workflow(user_query)
     history[-1] = (user_query, response)
 
-    # Format references for display with text passages
-    formatted_references = ""
-    docs = query_chroma(vectorstore, user_query, k=5)
-    for i, (doc, ref) in enumerate(docs):
-        formatted_references += f"""
-        <div style="border: 1px solid #ddd; padding: 10px; margin-bottom: 10px; border-radius: 5px;">
-            <h3 style="margin-top: 0;">Reference {i+1}</h3>
-            <p><strong>Source:</strong> {ref}</p>
-            <button onclick="var elem = document.getElementById('text-{i}'); var button = this; if (elem.style.display === 'block') {{ elem.style.display = 'none'; button.innerHTML = '&#9654; show source text'; }} else {{ elem.style.display = 'block'; button.innerHTML = '&#9660; hide source text'; }}">{{'&#9654; show source text'}}</button>
-            <div id="text-{i}" style="display: none;">
-                <p><strong>Text:</strong> {doc}</p>
-            </div>
-        </div>
-        """
-
-    yield history, formatted_references    
+    yield history  
 
 def main():
     with gr.Blocks() as demo:
@@ -447,14 +432,10 @@ def main():
                         examples_per_page=3,
                     )
 
-                with gr.Column(scale=3):
-                    with gr.Tab("References"):
-                        doc_citation = gr.HTML("<p>References used in answering the question will be displayed below.</p>")
-
-            #user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
-            #submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
-            user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
-            submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
+            user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
+            submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot])
+            #user_txt.submit(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
+            #submit_btn.click(check_input_text, user_txt, None).success(add_text, [chatbot, user_txt], [chatbot, user_txt]).then(bot_kadi, [chatbot], [chatbot, doc_citation])
             clear_btn.click(lambda: None, None, chatbot, queue=False)
 
     demo.launch()