Update app.py

app.py

@@ -75,44 +75,6 @@ def get_response(history, query):
         app.chat_history.append((query, "I have no information about it. Feed me knowledge, please!"))
         return history, f"I have no information about it. Feed me knowledge, please! Error: {str(e)}"
 
-# Function to get response for the current RAG tab
-def get_response_current(history, query):
-    if app.chain is None:
-        raise gr.Error("The chain has not been built yet. Please ensure the vector database is built before querying.")
-    
-    try:
-        result = app.chain.invoke(
-            {"question": query, "chat_history": app.chat_history}
-        )
-        app.chat_history.append((query, result["answer"]))
-        source_docs = result["source_documents"]
-        source_texts = []
-        for doc in source_docs:
-            source_texts.append(f"Page {doc.metadata['page'] + 1}: {doc.page_content}")
-        source_texts_str = "\n\n".join(source_texts)
-        history[-1] = (history[-1][0], result["answer"])
-        return history, source_texts_str
-    except Exception as e:
-        app.chat_history.append((query, "I have no information about it. Feed me knowledge, please!"))
-        return history, f"I have no information about it. Feed me knowledge, please! Error: {str(e)}"
-
-# Function to render file
-def render_file(file) -> Image.Image:
-    doc = fitz.open(file.name)
-    page = doc[0]
-    pix = page.get_pixmap(dpi=150)
-    image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-    return image
-
-# Function to purge chat and render first page of PDF
-def purge_chat_and_render_first(file) -> Image.Image:
-    app.chat_history = []
-    doc = fitz.open(file.name)
-    page = doc[0]
-    pix = page.get_pixmap(dpi=150)
-    image = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
-    return image
-
 # Function to refresh chat
 def refresh_chat():
     app.chat_history = []
@@ -167,49 +129,45 @@ with gr.Blocks() as demo:
         outputs=[api_key_status]
     )           
 
-    with gr.Tab("Take a Dialectical Behaviour Therapy with Me"):
-        with gr.Column():
-            chatbot_current = gr.Chatbot(elem_id="chatbot_current")
-            txt_current = gr.Textbox(
-                show_label=False,
-                placeholder="Enter text and press submit",
-                scale=2
-            )
-            submit_btn_current = gr.Button("Submit", scale=1)
-            refresh_btn_current = gr.Button("Refresh Chat", scale=1)
-            source_texts_output_current = gr.Textbox(label="Source Texts", interactive=False)
-
-            submit_btn_current.click(
-                fn=add_text,
-                inputs=[chatbot_current, txt_current],
-                outputs=[chatbot_current],
-                queue=False,
-            ).success(
-                fn=get_response_current, inputs=[chatbot_current, txt_current], outputs=[chatbot_current, source_texts_output_current]
-            )
-
-            refresh_btn_current.click(
-                fn=refresh_chat,
-                inputs=[],
-                outputs=[chatbot_current],
-            )
+    chatbot_current = gr.Chatbot(elem_id="chatbot_current")
+    txt_current = gr.Textbox(
+        show_label=False,
+        placeholder="Enter text and press submit",
+        scale=2
+    )
+    submit_btn_current = gr.Button("Submit", scale=1)
+    refresh_btn_current = gr.Button("Refresh Chat", scale=1)
+    source_texts_output_current = gr.Textbox(label="Source Texts", interactive=False)
+
+    submit_btn_current.click(
+        fn=add_text,
+        inputs=[chatbot_current, txt_current],
+        outputs=[chatbot_current],
+        queue=False,
+    ).success(
+        fn=get_response, inputs=[chatbot_current, txt_current], outputs=[chatbot_current, source_texts_output_current]
+    )
+
+    refresh_btn_current.click(
+        fn=refresh_chat,
+        inputs=[],
+        outputs=[chatbot_current],
+    )
+
+    question_dropdown = gr.Dropdown(
+        label="Select an example question",
+        choices=questions
+    )
+    question_submit_btn = gr.Button("Submit Question")
     
-    with gr.Tab("Example Questions"):
-        gr.Markdown("### Example Questions")
-        question_dropdown = gr.Dropdown(
-            label="Select a question",
-            choices=questions
-        )
-        question_submit_btn = gr.Button("Submit Question")
-        
-        question_submit_btn.click(
-            fn=add_text,
-            inputs=[chatbot_current, question_dropdown],
-            outputs=[chatbot_current],
-            queue=False,
-        ).success(
-            fn=get_response_current, inputs=[chatbot_current, question_dropdown], outputs=[chatbot_current, source_texts_output_current]
-        )
+    question_submit_btn.click(
+        fn=add_text,
+        inputs=[chatbot_current, question_dropdown],
+        outputs=[chatbot_current],
+        queue=False,
+    ).success(
+        fn=get_response, inputs=[chatbot_current, question_dropdown], outputs=[chatbot_current, source_texts_output_current]
+    )
 
 demo.queue()
 demo.launch()