Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
b048432
1
Parent(s):
edc6083
New app
Browse files- app.py +2 -2
- evaluations/repo_evaluations.py +12 -4
- evaluations/utils.py +2 -2
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
|
6 |
|
7 |
-
|
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 |
-
|
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,
|
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(
|
51 |
results["pred_live"] = "No"
|
52 |
return results
|
53 |
|
54 |
-
zip = zipfile.ZipFile(
|
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
|