Update app.py

app.py

@@ -71,41 +71,6 @@ df = df.drop(['message', 'file'], axis = 1)
 import chromadb
 chroma_client = chromadb.Client()
 
-collection = chroma_client.create_collection(name="emails")
-
-df.loc[4, 'text']
-
-for i in df.index:
-    collection.add(
-     
-        documents = df.loc[i, 'text'],
-
-      
-        metadatas = [{"sender": df.loc[i, 'sender'],
-                     "recipient1": df.loc[i, 'recipient1'],
-                     "recipient2": df.loc[i, 'recipient2'],
-                     "recipient3": df.loc[i, 'recipient3'],
-                     "subject": df.loc[i, 'Subject'],
-                     "folder": df.loc[i, 'folder'],
-                     "date": str(df.loc[i, 'date'])
-                     }],
-
-     
-        ids = str(i)
-    )
-
-collection.get(
-    ids=["140"]
-)
-
-results = collection.query(
-    query_texts = ["this is a document"],
-    n_results = 2,
-    include = ['distances', 'metadatas', 'documents']
-)
-results
-
-
 from chromadb.utils import embedding_functions
 
 
@@ -148,25 +113,6 @@ results
 import gradio as gr
 
 
-def query_chromadb(question,numberOfResults):
-    results = collection_minilm.query(
-          n_results = numberOfResults,
-      )
-
-    return results['documents'][0]
-
-iface = gr.Interface(
-    fn=query_chromadb,
-    inputs=["text","number"],
-    outputs="text",
-    title="Email Dataset Interface",
-    description="Insert the question or the key word to find the topic correlated in the dataset"
-)
-
-iface.launch(share=True)
-
-
-
 import ast
 
 def create_output(dictionary, number):
@@ -218,7 +164,6 @@ def query_chromadb_advanced(question,numberOfResults):
 
 result_advance = query_chromadb_advanced("bank", 4)
 
-print(result_advance)
 
 iface = gr.Interface(
     fn=query_chromadb_advanced,