Update app.py

app.py

@@ -3,8 +3,6 @@ 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
-from llama_index.llms.huggingface import HuggingFaceInferenceAPI
-from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 import random
 import datetime
 
@@ -89,6 +87,18 @@ 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 your Gradio chat interface function
 def chat_interface(message, history):
     cookies = {}  # You might need to get cookies from the request in a real implementation
@@ -132,7 +142,10 @@ div.svelte-1rjryqp{display: none;}
 div.progress-text.svelte-z7cif2.meta-text {display: none;}
 '''
 
-# Use Gradio Blocks to wrap components
-chat = gr.ChatInterface(chat_interface, css=css, clear_btn=None, undo_btn=None, retry_btn=None).launch()
+# 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)
 
 # Launch the Gradio interface
+demo.launch()