Aiswarya Sankar commited on
Commit
2b9e343
·
1 Parent(s): 2ea0c80

Clean code

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -75,12 +75,11 @@ repo_name = gr.State()
75
  git_tickets = gr.State()
76
  git_titles = gr.State()
77
  git_ticket_choices = gr.State()
 
78
  git_tickets.value = []
79
  git_titles.value = []
80
  git_ticket_choices.value = []
81
 
82
- # repoName = "https://github.com/gradio-app/gradio.git"
83
-
84
  embeddings = OpenAIEmbeddings(disallowed_special=())
85
 
86
  def git_clone(repo_url):
@@ -179,6 +178,8 @@ def index_repo(textbox: str, dropdown: str) -> Response:
179
  stdout="",
180
  )
181
 
 
 
182
  return {
183
  success_response: "SUCCESS",
184
  launch_product: gr.update(visible=True)
@@ -194,7 +195,7 @@ def answer_questions(question: str, github: str, **kwargs) -> Response:
194
  try:
195
  embeddings = OpenAIEmbeddings(disallowed_special=())
196
  pathName = github.split('/')[-1]
197
- dataset_path = "hub://aiswaryas/" + pathName
198
 
199
  print("before reading repo")
200
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
@@ -321,7 +322,7 @@ def generateDocumentationPerFolder(dir, github):
321
  try:
322
  embeddings = OpenAIEmbeddings(disallowed_special=())
323
  pathName = github.split('/')[-1]
324
- dataset_path = "hub://aiswaryas/" + pathName
325
 
326
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
327
 
@@ -371,7 +372,7 @@ def solveGithubIssue(ticket, history) -> Response:
371
  try:
372
  embeddings = OpenAIEmbeddings(disallowed_special=())
373
  pathName = github.split('/')[-1]
374
- dataset_path = "hub://aiswaryas/" + pathName
375
 
376
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding=embeddings)
377
 
@@ -423,7 +424,7 @@ def bot(history, **kwargs):
423
  try:
424
  embeddings = OpenAIEmbeddings(disallowed_special=())
425
  pathName = github.split('/')[-1]
426
- dataset_path = "hub://aiswaryas/" + pathName
427
 
428
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
429
  retriever = db.as_retriever()
 
75
  git_tickets = gr.State()
76
  git_titles = gr.State()
77
  git_ticket_choices = gr.State()
78
+ vector_db_url = gr.State()
79
  git_tickets.value = []
80
  git_titles.value = []
81
  git_ticket_choices.value = []
82
 
 
 
83
  embeddings = OpenAIEmbeddings(disallowed_special=())
84
 
85
  def git_clone(repo_url):
 
178
  stdout="",
179
  )
180
 
181
+ vector_db_url.value = dataset_path
182
+
183
  return {
184
  success_response: "SUCCESS",
185
  launch_product: gr.update(visible=True)
 
195
  try:
196
  embeddings = OpenAIEmbeddings(disallowed_special=())
197
  pathName = github.split('/')[-1]
198
+ dataset_path = vector_db_url.value
199
 
200
  print("before reading repo")
201
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
 
322
  try:
323
  embeddings = OpenAIEmbeddings(disallowed_special=())
324
  pathName = github.split('/')[-1]
325
+ dataset_path = vector_db_url.value
326
 
327
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
328
 
 
372
  try:
373
  embeddings = OpenAIEmbeddings(disallowed_special=())
374
  pathName = github.split('/')[-1]
375
+ dataset_path = vector_db_url.value
376
 
377
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding=embeddings)
378
 
 
424
  try:
425
  embeddings = OpenAIEmbeddings(disallowed_special=())
426
  pathName = github.split('/')[-1]
427
+ dataset_path = vector_db_url.value
428
 
429
  db = DeepLake(dataset_path=dataset_path, read_only=True, embedding_function=embeddings)
430
  retriever = db.as_retriever()