Skip to content

Commit

Permalink
Modified Dynamic Path for Model View
Browse files Browse the repository at this point in the history
  • Loading branch information
Shanks0465 committed Aug 28, 2024
1 parent 659e670 commit c4a83d9
Show file tree
Hide file tree
Showing 3 changed files with 93 additions and 56 deletions.
76 changes: 49 additions & 27 deletions frontend/components/Models.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -7,18 +7,11 @@ import {
Box,
Heading,
Text,
Button,
Image,
Icon,
IconButton,
createIcon,
IconProps,
useColorModeValue,
Link,
HStack,
} from "@chakra-ui/react";
import { useQuery } from "react-query";
import { API_URL, LANGUAGE_CODE_NAMES } from "@/app/config";
import { API_URL } from "@/app/config";
import { useEffect, useState } from "react";
import axios from "axios";
import NMT from "./TryOut/NMT";
Expand All @@ -34,7 +27,38 @@ const fetchModel = async ({ title }: { title: string }) => {
}
};

export default function ModelView({ slug }: { slug: Array<string> }) {
interface Model {
service_id: string;
inferenceSchema: any;
languageFilters: any;
hfData: any;
conference: string;
paper_link: string | undefined;
github_link: string | undefined;
title: string;
description?: string;
}

const renderTryOut = ({ area, model }: { area: string; model: Model }) => {
switch (area) {
case "NMT":
return (
<NMT
sourceLanguages={model.languageFilters.sourceLanguages}
targetLanguages={model.languageFilters.targetLanguages}
serviceId={model.service_id}
/>
);
}
};

export default function ModelView({
area,
title,
}: {
area: string;
title: string;
}) {
const [model, setModel] = useState<{
service_id: string;
inferenceSchema: any;
Expand All @@ -57,13 +81,11 @@ export default function ModelView({ slug }: { slug: Array<string> }) {
service_id: "",
});

const [hfData, setHFData] = useState({});

const {
isLoading: modelLoading,
error: modelError,
data: modelData,
} = useQuery(["fetchModel", slug], () => fetchModel({ title: slug[1] }));
} = useQuery(["fetchModel", title], () => fetchModel({ title: title }));

useEffect(() => {
if (modelError || modelLoading) {
Expand Down Expand Up @@ -112,16 +134,20 @@ export default function ModelView({ slug }: { slug: Array<string> }) {
Conference : {model.conference}
</Text>
</Box>
<Box
borderRadius={15}
p={1}
borderWidth={3}
borderColor={"a4borange"}
>
<Text textColor={"a4borange"}>
Downloads : {model.hfData.downloads}
</Text>
</Box>
{model.hfData.downloads ? (
<Box
borderRadius={15}
p={1}
borderWidth={3}
borderColor={"a4borange"}
>
<Text textColor={"a4borange"}>
Downloads : {model.hfData.downloads}
</Text>
</Box>
) : (
<></>
)}
</HStack>
<Text color={"gray.500"}>{model.description}</Text>
<HStack>
Expand Down Expand Up @@ -150,11 +176,7 @@ export default function ModelView({ slug }: { slug: Array<string> }) {
position={"relative"}
w={"full"}
>
<NMT
sourceLanguages={model.languageFilters.sourceLanguages}
targetLanguages={model.languageFilters.targetLanguages}
serviceId={model.service_id}
/>
{renderTryOut({ area: area, model: model })}
</Flex>
</Stack>
</Container>
Expand Down
29 changes: 0 additions & 29 deletions frontend/src/app/areas/model/[...slug]/page.tsx

This file was deleted.

44 changes: 44 additions & 0 deletions frontend/src/app/areas/model/[area]/[title]/page.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
import { title } from "process";
import ModelView from "../../../../../../components/Models";

interface ParamsType {
slug: Array<string>;
}

interface Model {
area: string;
title: string;
}

import axios from "axios";

export async function generateStaticParams() {
const response = await axios.get(
"https://admin.models.ai4bharat.org/models/"
);
const models = response.data;

let params: any[] = [];

models.forEach((model: Model) => {
params.push({
area: model.area,
title: model.title,
});
});

return params;
}

export const dynamicParams = true;
export default function Model({
params,
}: {
params: { area: string; title: string };
}) {
return (
<>
<ModelView area={params.area} title={params.title} />
</>
);
}

0 comments on commit c4a83d9

Please sign in to comment.