Update app.py

app.py

@@ -3,10 +3,11 @@ import os
 from http.cookies import SimpleCookie
 from dotenv import load_dotenv
 from llama_index.core import StorageContext, load_index_from_storage, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, Settings
-import random
-import datetime
 from llama_index.llms.huggingface import HuggingFaceInferenceAPI
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+import random
+import datetime
+import webbrowser  # To handle the redirect
 
 # Load environment variables
 load_dotenv()
@@ -89,17 +90,13 @@ def handle_query(query, cookies=None):
 
     return response
 
-# Function to detect iframe and block chat history access
-def detect_iframe():
-    iframe_script = '''
-        <script>
-            if (window != window.top) {
-                alert("Chat history access is disabled in iframes.");
-                document.getElementById('chat_history').style.display = 'none';
-            }
-        </script>
-    '''
-    return iframe_script
+# Define the button click function
+def retrieve_history_and_redirect():
+    # Here you can retrieve the chat history if necessary
+    # You can return it, log it, or perform any necessary action
+    # Redirect to the specified page
+    webbrowser.open("https://redfernstech.com/chat-bot-test")
+    return "Redirecting..."
 
 # Define your Gradio chat interface function
 def chat_interface(message, history):
@@ -144,10 +141,15 @@ div.svelte-1rjryqp{display: none;}
 div.progress-text.svelte-z7cif2.meta-text {display: none;}
 '''
 
-# Use Gradio Blocks to wrap components and add iframe detection
-with gr.Blocks() as demo:
-    gr.HTML(detect_iframe())
-    chat = gr.ChatInterface(chat_interface, css=css, clear_btn=None, undo_btn=None, retry_btn=None)
+# Use Gradio Blocks to wrap components
+with gr.Blocks(css=css) as demo:
+    chat = gr.ChatInterface(chat_interface, clear_btn=None, undo_btn=None, retry_btn=None)
+    
+    # Button to retrieve history and redirect
+    redirect_button = gr.Button("Retrieve History & Redirect")
+    
+    # Connect the button with the function
+    redirect_button.click(fn=retrieve_history_and_redirect, inputs=[], outputs=chat)
 
 # Launch the Gradio interface
 demo.launch()