Update app.py

app.py

@@ -15,6 +15,7 @@ if torch.cuda.is_available():
 validation_results = json.load(open('validation_results.json'))
 scores, thresholds, precisions, recalls = validation_results['scores'], validation_results['thresholds'], validation_results['precisions'], validation_results['recalls']
 
+
 def get_threshold_precision(score_):
     for score, threshold, precision, recall in zip(scores, thresholds, precisions, recalls):
         if score_ < score:
@@ -24,6 +25,34 @@ def get_threshold_precision(score_):
         prev_threshold = score_
     return prev_threshold, prev_precision, prev_recall
 
+def extract_arxiv_id(input_text):
+    if 'arxiv.org' in input_text:
+        parsed_url = urlparse(input_text)
+        query = parse_qs(parsed_url.query)
+        path_parts = parsed_url.path.split('/')
+        if 'id_list' in query:
+            return query['id_list'][0]
+        elif path_parts[-2] in ['abs', 'pdf']:
+            return path_parts[-1].replace('.pdf', '')
+    return input_text
+
+def fetch_arxiv_data(arxiv_id):
+    time.sleep(3)  # Comply with arXiv API terms of usage
+    query_url = f'http://export.arxiv.org/api/query?id_list={arxiv_id}'
+    response = feedparser.parse(query_url)
+    if response.entries:
+        entry = response.entries[0]
+        title = entry.title
+        authors = ', '.join(author.name for author in entry.authors)
+        abstract = entry.summary
+        return title, authors, abstract
+    return "", "", ""
+
+def update_fields(url_or_id):
+    arxiv_id = extract_arxiv_id(url_or_id)
+    title, authors, abstract = fetch_arxiv_data(arxiv_id)
+    return title, authors, abstract
+
 def normalize_spaces(text):
     return re.sub(r'\s+', ' ', text).strip()
 
@@ -69,10 +98,14 @@ example_abstract = "Chatbots such as GPT-4 and ChatGPT are now serving millions
     
 iface = gr.Interface(
     fn=predict,
-    inputs=[gr.Textbox(placeholder="Enter paper title", value=example_title), gr.Textbox(placeholder="Enter authors (separated by comma)", value=example_authors), gr.TextArea(placeholder="Enter abstract", value=example_abstract)],
+    inputs=[gr.Textbox(label="Paper Title", placeholder="Enter paper title", value=example_title), 
+            gr.Textbox(label="Authors (separated by comma)", placeholder="Enter authors (separated by comma)", value=example_authors), 
+            gr.TextArea(label="Abstract", placeholder="Enter abstract", value=example_abstract),
+            gr.Textbox(label="[Optional] Autofill using arXiv URL/ID", placeholder="[Optional] Autofill using arXiv URL/ID", on_change=update_fields, change_elements=["title", "authors", "abstract"]),
+],
     outputs=[gr.Textbox(label="Predicted Score"), gr.Textbox(label="Predicted Selection Probability")],
     title="Paper Selection Prediction",
-    description="Predict if @_akhaliq will select your paper. Enter the title, authors, and abstract of your paper.",
+    description="Predict if @_akhaliq will select your paper. Enter the title, authors, and abstract of your paper, or enter an arXiv URL/ID.",
     live=False,
 )