From 8e0bc8eb98343b69890810194f65fca6a51a4a27 Mon Sep 17 00:00:00 2001
From: cdeng65 <cdeng65@gatech.edu>
Date: Tue, 25 Jul 2023 22:01:47 -0400
Subject: [PATCH] added docs

---
 src/screens/ARB/Arb.tsx                       | 28 ++++++++++++++++---
 .../mongodb/actions/numericalProblem.js       |  6 ++--
 2 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/src/screens/ARB/Arb.tsx b/src/screens/ARB/Arb.tsx
index 9564601..c3d44fb 100644
--- a/src/screens/ARB/Arb.tsx
+++ b/src/screens/ARB/Arb.tsx
@@ -16,6 +16,7 @@ import {
   FaArrowCircleRight,
   FaFilePdf,
   FaGithub,
+  FaBook,
   FaImages,
 } from "react-icons/fa";
 import { AiFillFile } from "react-icons/ai";
@@ -143,11 +144,12 @@ const Arb: React.FC = () => {
             <Flex mt={4} wrap="wrap" justify="center">
               <Link
                 href="https://arxiv.org/pdf/2307.13692.pdf"
-                w={["100%", "auto"]}
+                w={["50%", "auto"]}
                 mb={4}
                 mr={[0, 4]}
               >
                 <Button
+                  w={"140px"}
                   leftIcon={<FaFilePdf />}
                   colorScheme="teal"
                   variant="solid"
@@ -157,11 +159,12 @@ const Arb: React.FC = () => {
               </Link>
               <Link
                 href="https://arxiv.org/abs/2307.13692"
-                w={["100%", "auto"]}
+                w={["50%", "auto"]}
                 mb={4}
                 mr={[0, 4]}
               >
                 <Button
+                  w={"140px"}
                   leftIcon={<AiFillFile />}
                   colorScheme="teal"
                   variant="solid"
@@ -171,11 +174,12 @@ const Arb: React.FC = () => {
               </Link>
               <Link
                 href="https://github.com/TheDuckAI/arb"
-                w={["100%", "auto"]}
+                w={["33.33%", "auto"]}
                 mb={4}
                 mr={[0, 4]}
               >
                 <Button
+                  w={"140px"}
                   leftIcon={<FaGithub />}
                   colorScheme="teal"
                   variant="solid"
@@ -183,8 +187,9 @@ const Arb: React.FC = () => {
                   Code
                 </Button>
               </Link>
-              <Link href="/home" w={["100%", "auto"]} mb={4}>
+              <Link href="/home" w={["33.33%", "auto"]} mb={4}>
                 <Button
+                  w={"140px"}
                   leftIcon={<FaImages />}
                   colorScheme="teal"
                   variant="solid"
@@ -192,6 +197,21 @@ const Arb: React.FC = () => {
                   Interface
                 </Button>
               </Link>
+              <Link
+                href="https://app.swaggerhub.com/apis-docs/arb-dataset/arb-api/1.0.5"
+                w={["33.33%", "auto"]}
+                mb={4}
+                mr={[0, 4]}
+              >
+                <Button
+                  w={"140px"}
+                  leftIcon={<FaBook />}
+                  colorScheme="teal"
+                  variant="solid"
+                >
+                  API
+                </Button>
+              </Link>
             </Flex>
           </Flex>
         </Container>
diff --git a/src/server/mongodb/actions/numericalProblem.js b/src/server/mongodb/actions/numericalProblem.js
index 45e51e7..5f23101 100644
--- a/src/server/mongodb/actions/numericalProblem.js
+++ b/src/server/mongodb/actions/numericalProblem.js
@@ -70,7 +70,7 @@ async function getAllPhysicsImgProblems() {
 
 async function findMathProblemById(id) {
   return findOneByMongoId(mathNumericalModel, id, {
-    "Problem Statement": 1,
+    Problem_Statement: 1,
     Solution: 1,
     "Final Answer": 1,
   });
@@ -78,7 +78,7 @@ async function findMathProblemById(id) {
 
 async function findPhysicsProblemById(id) {
   return findOneByMongoId(physicsNumericalModel, id, {
-    "Problem Statement": 1,
+    Problem_Statement: 1,
     Solution: 1,
     "Final Answer": 1,
   });
@@ -86,7 +86,7 @@ async function findPhysicsProblemById(id) {
 
 async function findPhysicsImgProblemById(id) {
   return findOneByMongoId(physicsImgModel, id, {
-    "Problem Statement": 1,
+    Problem_Statement: 1,
     Images: 1,
     Solution: 1,
     "Final Answer": 1,