update

app.py

@@ -32,6 +32,7 @@ gtag('config', 'G-EZ77X5T529');
 </script>
 """
 
+
 def replace_texttt(text):
     return re.sub(r"\\texttt\{(.*?)\}", r"*\1*", text)
 
@@ -54,8 +55,13 @@ def get_paper_info(paper_id):
 
 
 def get_paper_from_huggingface(paper_id):
+    top_level_paper_id = paper_id.split(".")[0]
+    path_in_repo = f"papers/{top_level_paper_id}/{paper_id}.tex"
+
     try:
-        url = f"https://huggingface.co/datasets/taesiri/arxiv_db/raw/main/papers/{paper_id}.tex"
+        url = (
+            f"https://huggingface.co/datasets/taesiri/arxiv_db/raw/main/{path_in_repo}"
+        )
         response = requests.get(url)
         response.raise_for_status()
         return response.text
@@ -166,9 +172,12 @@ def load_context(paper_id):
             if os.path.getsize(temp_file_path) > 1:
                 hf_api = HfApi(token=os.environ["HUGGINGFACE_TOKEN"])
 
+                top_level_paper_id = paper_id.split(".")[0]
+                path_in_repo = f"papers/{top_level_paper_id}/{paper_id}.tex"
+
                 hf_api.upload_file(
                     path_or_fileobj=temp_file_path,
-                    path_in_repo=f"papers/{paper_id}.tex",
+                    path_in_repo=path_in_repo,
                     repo_id="taesiri/arxiv_db",
                     repo_type="dataset",
                 )