attilasimko commited on
Commit
b048432
·
1 Parent(s): edc6083
app.py CHANGED
@@ -2,9 +2,9 @@ import streamlit as st
2
  from evaluations.repo_evaluations import evaluate
3
 
4
 
5
- repo_link = st.text_input("Github repository link:", value="", type="default", help=None, on_change=eval)
6
 
7
- def eval():
8
  evaluate(llm=None, verbose=2, repo_url=repo_link)
9
 
10
  with st.form("my_form"):
 
2
  from evaluations.repo_evaluations import evaluate
3
 
4
 
5
+ repo_link = st.text_input("Github repository link:", value="", type="default", help=None)
6
 
7
+ if (repo_link):
8
  evaluate(llm=None, verbose=2, repo_url=repo_link)
9
 
10
  with st.form("my_form"):
evaluations/repo_evaluations.py CHANGED
@@ -20,7 +20,7 @@ def init_llm(verbose):
20
  log(verbose, "LOG", "Initializing LLM...")
21
 
22
  def evaluate(llm, verbose, repo_url, title=None, year=None):
23
- repo_name = "data/repo.zip"
24
  token = os.getenv("githubToken")
25
  # token = userdata.get('githubToken')
26
 
@@ -34,10 +34,18 @@ def evaluate(llm, verbose, repo_url, title=None, year=None):
34
  try:
35
  if (get_api_link(repo_url) != ""):
36
  results["pred_valid"] = True
 
 
 
 
 
 
 
 
37
  username, repo_name = decompose_url(repo_url)
38
  log(verbose, "LOG", f"Fetching github repository: https://github.com/{username}/{repo_name}")
39
 
40
- fetch_repo(verbose, repo_url, repo_name, token)
41
 
42
  if ((title != None) & (year != None) & (title != "") & (year != "")):
43
  res = fetch_openalex(verbose, title, year)
@@ -47,11 +55,11 @@ def evaluate(llm, verbose, repo_url, title=None, year=None):
47
  res = res[0]
48
  results["pred_citations"] = res["cited_by_count"]
49
 
50
- if (not(os.path.exists(repo_name))):
51
  results["pred_live"] = "No"
52
  return results
53
 
54
- zip = zipfile.ZipFile(repo_name)
55
  readme = fetch_readme(zip)
56
  results["pred_stars"] = fetch_repo_stars(verbose, repo_url, token)
57
 
 
20
  log(verbose, "LOG", "Initializing LLM...")
21
 
22
  def evaluate(llm, verbose, repo_url, title=None, year=None):
23
+ repository_zip_name = "data/repo.zip"
24
  token = os.getenv("githubToken")
25
  # token = userdata.get('githubToken')
26
 
 
34
  try:
35
  if (get_api_link(repo_url) != ""):
36
  results["pred_valid"] = True
37
+ else:
38
+ results["pred_live"] = "No"
39
+ results["pred_training"] = "No"
40
+ results["pred_evaluation"] = "No"
41
+ results["pred_weights"] = "No"
42
+ results["pred_packages"] = "No"
43
+ return results
44
+
45
  username, repo_name = decompose_url(repo_url)
46
  log(verbose, "LOG", f"Fetching github repository: https://github.com/{username}/{repo_name}")
47
 
48
+ fetch_repo(verbose, repo_url, repository_zip_name, token)
49
 
50
  if ((title != None) & (year != None) & (title != "") & (year != "")):
51
  res = fetch_openalex(verbose, title, year)
 
55
  res = res[0]
56
  results["pred_citations"] = res["cited_by_count"]
57
 
58
+ if (not(os.path.exists(repository_zip_name))):
59
  results["pred_live"] = "No"
60
  return results
61
 
62
+ zip = zipfile.ZipFile(repository_zip_name)
63
  readme = fetch_readme(zip)
64
  results["pred_stars"] = fetch_repo_stars(verbose, repo_url, token)
65
 
evaluations/utils.py CHANGED
@@ -34,7 +34,7 @@ def decompose_url(url):
34
  repo_name = parts[1]
35
  return username, repo_name
36
  except:
37
- return ""
38
 
39
 
40
  def fetch_repo_stars(verbose, repo_url, token):
@@ -119,7 +119,7 @@ def log(verbose, log_type, log_text, hf=False):
119
  if (log_text.startswith("\n")):
120
  print("\n")
121
  log_text = log_text.lstrip('\n')
122
-
123
  if (verbose == 1):
124
  print(log_text)
125
  return
 
34
  repo_name = parts[1]
35
  return username, repo_name
36
  except:
37
+ return None, None
38
 
39
 
40
  def fetch_repo_stars(verbose, repo_url, token):
 
119
  if (log_text.startswith("\n")):
120
  print("\n")
121
  log_text = log_text.lstrip('\n')
122
+
123
  if (verbose == 1):
124
  print(log_text)
125
  return