Kratagya101 commited on
Commit
89ab7c2
·
1 Parent(s): 1c36cb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -36
app.py CHANGED
@@ -1,38 +1,16 @@
1
- from gpt_index import SimpleDirectoryReader, GPTListIndex, GPTSimpleVectorIndex, LLMPredictor, PromptHelper
2
- from langchain.chat_models import ChatOpenAI
3
- import gradio as gr
4
- import sys
5
  import os
6
 
7
- os.environ["OPENAI_API_KEY"] = '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'
8
-
9
- def construct_index(directory_path):
10
- max_input_size = 4096
11
- num_outputs = 512
12
- max_chunk_overlap = 20
13
- chunk_size_limit = 600
14
-
15
- prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
16
-
17
- llm_predictor = LLMPredictor(llm=ChatOpenAI(temperature=0.7, model_name="gpt-3.5-turbo", max_tokens=num_outputs))
18
-
19
- documents = SimpleDirectoryReader(directory_path).load_data()
20
-
21
- index = GPTSimpleVectorIndex(documents, llm_predictor=llm_predictor, prompt_helper=prompt_helper)
22
-
23
- index.save_to_disk('index.json')
24
-
25
- return index
26
-
27
- def chatbot(input_text):
28
- index = GPTSimpleVectorIndex.load_from_disk('index.json')
29
- response = index.query(input_text, response_mode="compact")
30
- return response.response
31
-
32
- iface = gr.Interface(fn=chatbot,
33
- inputs=gr.components.Textbox(lines=7, label="Enter your text"),
34
- outputs="text",
35
- title="Custom-trained AI Chatbot")
36
-
37
- index = construct_index("docs")
38
- iface.launch()
 
 
 
 
 
1
  import os
2
 
3
+ def delete_files_in_folder(folder_path):
4
+ try:
5
+ for filename in os.listdir(folder_path):
6
+ file_path = os.path.join(folder_path, filename)
7
+ if os.path.isfile(file_path):
8
+ os.remove(file_path)
9
+ print(f"Deleted: {filename}")
10
+ print("All files deleted successfully.")
11
+ except Exception as e:
12
+ print(f"An error occurred: {e}")
13
+
14
+ if __name__ == "__main__":
15
+ folder_to_clear = "docs" # Change this to your actual folder path
16
+ delete_files_in_folder(folder_to_clear)