TheBobBob commited on
Commit
0a2169d
·
verified ·
1 Parent(s): 1d3b85c

Delete main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -45
main.py DELETED
@@ -1,45 +0,0 @@
1
- from createVectorDB import createVectorDB
2
- from splitBioModels import splitBioModels
3
- from createDocuments import createDocuments
4
- from generateResponse import generateResponse
5
- from selectBioModels import search_biomodels
6
- from selectBioModels import copy_matching_files
7
-
8
- DATA_PATH = r"C:\Users\navan\Downloads\BioModelsRAG\BioModelsRAG\2data"
9
- CHROMA_DATA_PATH = r"C:\Users\navan\Downloads\BioModelsRAG\CHROMA_EMBEDDINGS_PATH"
10
- directory = r'C:\Users\navan\Downloads\BioModelsRAG\BioModelsRAG\data'
11
- output_file = r'C:\Users\navan\Downloads\BioModelsRAG\biomodels_output.csv'
12
- final_models_folder = r'C:\Users\navan\Downloads\BioModelsRAG\final_models'
13
- user_keywords = input("Keyword you would like to search for: ").split()
14
-
15
-
16
- def main(report:bool = True, directory = DATA_PATH, chroma_data_path = CHROMA_DATA_PATH):
17
- data = []
18
- search_biomodels(directory, user_keywords, output_file)
19
- copy_matching_files(output_file, directory, final_models_folder)
20
-
21
- splitBioModels(directory=DATA_PATH, final_items=data)
22
-
23
- collection = createVectorDB(
24
- collection_name="123456789101112131415",
25
- chroma_data_path=chroma_data_path,
26
- embed_model="all-MiniLM-L6-v2",
27
- metadata={"hnsw:space": "cosine"}
28
- )
29
-
30
- if report:
31
- print("Collection created:", collection)
32
-
33
- createDocuments(final_items=data, collection=collection)
34
-
35
- if report:
36
- print("Documents added to collection.")
37
-
38
- query = "What protein interacts with DesensitizedAch2?"
39
- result = generateResponse(query_text=query, collection=collection)
40
- return result
41
- #name of the program running v
42
- if __name__ == "__main__":
43
- result = main()
44
- print(result)
45
-