Update app.py

app.py

@@ -155,6 +155,13 @@ def get_relevant_context(query, top_k=5):
     context = "\n".join([result['metadata']['text'] for result in sorted_results])
     return context, sorted_results
 
+def truncate_context(context, max_tokens):
+    enc = encoding_for_model("gpt-4")
+    encoded = enc.encode(context)
+    if len(encoded) > max_tokens:
+        return enc.decode(encoded[:max_tokens])
+    return context
+
 def chat_with_ai(message):
     print(f"Chatting with AI, message: {message}")
     context, results = get_relevant_context(message)
@@ -193,12 +200,6 @@ def chat_with_ai(message):
     
     return ai_response, sources
 
-def clear_database():
-    print("Clearing database...")
-    index.delete(delete_all=True)
-    print("Database cleared")
-    return "Database cleared successfully."
-
 def process_youtube_links(links):
     results = []
     for link in links:
@@ -248,10 +249,6 @@ with col1:
                 st.warning("No content uploaded. Please provide at least one input.")
         st.session_state.upload_progress.empty()
 
-    if st.button("Clear Database"):
-        result = clear_database()
-        st.success(result)
-
 with col2:
     st.header("Chat")
     user_input = st.text_input("Ask a question about the uploaded content:")