From 4fe74efeb59b5c701d1d3af09f468f254304dc2f Mon Sep 17 00:00:00 2001
From: Siddharth Bramhecha <101277141+Sid-80@users.noreply.github.com>
Date: Fri, 19 Jul 2024 12:16:05 +0530
Subject: [PATCH] Feat/377 (#388)

* feat: Multi-Step Form

* feat: changes ui

* feat: Gemini Integaretion
---
 .env.example                   |  2 ++
 components/CoverLetterForm.tsx | 29 +++++++++++++++---
 components/cover-letter.tsx    |  7 +++++
 package-lock.json              |  9 ++++++
 package.json                   |  1 +
 utils/Gemini.ts                | 54 ++++++++++++++++++++++++++++++++++
 6 files changed, 98 insertions(+), 4 deletions(-)
 create mode 100644 utils/Gemini.ts

diff --git a/.env.example b/.env.example
index e8065af..3948e79 100644
--- a/.env.example
+++ b/.env.example
@@ -4,6 +4,8 @@ NEXT_PUBLIC_URL=https://git-re.vercel.app/
 NEXT_PUBLIC_UPSTASH_REDIS_URL=
 NEXT_PUBLIC_UPSTASH_REDIS_TOKEN=
 
+NEXT_PUBLIC_GEMINI_API_KEY=your-gemini-api-key
+
 # Generate the token from https://github.com/settings/tokens?type=beta
 # Repository Access to public repositories (read-only)
 NEXT_PUBLIC_GITHUB_TOKEN=
diff --git a/components/CoverLetterForm.tsx b/components/CoverLetterForm.tsx
index 4eccb55..d507951 100644
--- a/components/CoverLetterForm.tsx
+++ b/components/CoverLetterForm.tsx
@@ -14,6 +14,7 @@ import {
 import { Textarea } from "./ui/textarea";
 import { ArrowRightIcon } from "@radix-ui/react-icons";
 import { SetStateAction } from "react";
+import { getCoverLetter } from "@/utils/Gemini";
 
 const formSchema = z.object({
   jobDescription: z
@@ -34,7 +35,10 @@ type Props = {
   isSubmit: boolean;
   setIsJobDescription: React.Dispatch<SetStateAction<boolean>>;
   setIsResumeDetails: React.Dispatch<SetStateAction<boolean>>;
+  setIsResponseGenerated: React.Dispatch<SetStateAction<boolean>>;
+  setIsError: React.Dispatch<SetStateAction<boolean>>;
   setIsSubmit: React.Dispatch<SetStateAction<boolean>>;
+  setResponse: React.Dispatch<SetStateAction<string>>;
 };
 
 export default function CoverLetterForm({
@@ -44,6 +48,9 @@ export default function CoverLetterForm({
   setIsJobDescription,
   setIsSubmit,
   setIsResumeDetails,
+  setIsResponseGenerated,
+  setIsError,
+  setResponse,
 }: Props) {
   const form = useForm<z.infer<typeof formSchema>>({
     resolver: zodResolver(formSchema),
@@ -75,11 +82,25 @@ export default function CoverLetterForm({
     }
   }
 
-  function onSubmit(values: z.infer<typeof formSchema>) {
+  const onSubmit = async (values: z.infer<typeof formSchema>) => {
     // Do something with the form values.
     // ✅ This will be type-safe and validated.
-    console.log(values);
-  }
+    const { jobDescription, project, skills, experience } = values;
+
+    try {
+      const res = await getCoverLetter({
+        jobDescription,
+        project,
+        skills,
+        experience,
+      });
+
+      console.log(res);
+    } catch (err) {
+      setIsError(true);
+    }
+  };
+
   return (
     <Form {...form}>
       <form
@@ -122,7 +143,7 @@ export default function CoverLetterForm({
               name="skills"
               render={({ field }) => (
                 <FormItem>
-                  <FormLabel>Projects</FormLabel>
+                  <FormLabel>Skills</FormLabel>
                   <FormControl>
                     <Textarea
                       className="h-[100px]"
diff --git a/components/cover-letter.tsx b/components/cover-letter.tsx
index b750803..8dbd4e0 100644
--- a/components/cover-letter.tsx
+++ b/components/cover-letter.tsx
@@ -15,6 +15,10 @@ export default function CoverLetter({ user }: Props) {
   const [isResume, setIsResume] = useState(false);
   const [isSubmit, setIsSubmit] = useState(false);
 
+  const [isResponseGenerated, setIsResponseGenerated] = useState(false);
+  const [isError, setIsError] = useState(false);
+  const [response, setResponse] = useState("");
+
   if (user === null) {
     router.push("/");
   }
@@ -61,11 +65,14 @@ export default function CoverLetter({ user }: Props) {
 
         <div className="flex flex-1 items-center justify-center">
           <CoverLetterForm
+            setIsError={setIsError}
+            setIsResponseGenerated={setIsResponseGenerated}
             isJobDescription={isJobDescription}
             isResumeDetails={isResume}
             isSubmit={isSubmit}
             setIsResumeDetails={setIsResume}
             setIsJobDescription={setIsJobDescription}
+            setResponse={setResponse}
             setIsSubmit={setIsSubmit}
           />
         </div>
diff --git a/package-lock.json b/package-lock.json
index e417265..4518ddd 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -8,6 +8,7 @@
       "name": "git-re",
       "version": "0.1.0",
       "dependencies": {
+        "@google/generative-ai": "^0.15.0",
         "@hookform/resolvers": "^3.8.0",
         "@radix-ui/react-accordion": "^1.2.0",
         "@radix-ui/react-avatar": "^1.1.0",
@@ -424,6 +425,14 @@
       "resolved": "https://registry.npmjs.org/@floating-ui/utils/-/utils-0.2.1.tgz",
       "integrity": "sha512-9TANp6GPoMtYzQdt54kfAyMmz1+osLlXdg2ENroU7zzrtflTLrrC/lgrIfaSe+Wu0b89GKccT7vxXA0MoAIO+Q=="
     },
+    "node_modules/@google/generative-ai": {
+      "version": "0.15.0",
+      "resolved": "https://registry.npmjs.org/@google/generative-ai/-/generative-ai-0.15.0.tgz",
+      "integrity": "sha512-zs37judcTYFJf1U7tnuqnh7gdzF6dcWj9pNRxjA5JTONRoiQ0htrRdbefRFiewOIfXwhun5t9hbd2ray7812eQ==",
+      "engines": {
+        "node": ">=18.0.0"
+      }
+    },
     "node_modules/@graphql-typed-document-node/core": {
       "version": "3.2.0",
       "resolved": "https://registry.npmjs.org/@graphql-typed-document-node/core/-/core-3.2.0.tgz",
diff --git a/package.json b/package.json
index fe14e5d..6681683 100644
--- a/package.json
+++ b/package.json
@@ -9,6 +9,7 @@
     "lint": "next lint"
   },
   "dependencies": {
+    "@google/generative-ai": "^0.15.0",
     "@hookform/resolvers": "^3.8.0",
     "@radix-ui/react-accordion": "^1.2.0",
     "@radix-ui/react-avatar": "^1.1.0",
diff --git a/utils/Gemini.ts b/utils/Gemini.ts
new file mode 100644
index 0000000..0015594
--- /dev/null
+++ b/utils/Gemini.ts
@@ -0,0 +1,54 @@
+import { GoogleGenerativeAI } from "@google/generative-ai";
+
+const MODEL_NAME = "gemini-1.5-pro";
+const API_KEY1 = process.env.NEXT_PUBLIC_GEMINI_API_KEY as string;
+
+const chatHistory = [
+  {
+    role: "user",
+    parts: [
+      {
+        text: `You are an expert in generating cover letter. Your task is to generate cover letter for given job description and resume details. If you are unable to generate then provide output as Information is insufficient!!`,
+      },
+    ],
+  },
+  {
+    role: "model",
+    parts: [{ text: "How can I help you ?" }],
+  },
+];
+
+type Props = {
+  jobDescription: string;
+  project: string;
+  skills: string;
+  experience: string;
+};
+
+export async function getCoverLetter({
+  jobDescription,
+  project,
+  skills,
+  experience,
+}: Props) {
+  if (!API_KEY1) throw new Error("Invalid Api Key!");
+
+  const genAI = new GoogleGenerativeAI(API_KEY1);
+  const model = genAI.getGenerativeModel({ model: MODEL_NAME });
+
+  model.generationConfig = {
+    temperature: 0.9,
+    topK: 1,
+    topP: 1,
+  };
+
+  const chat = model.startChat({
+    history: chatHistory,
+  });
+
+  const result = await chat.sendMessage(
+    `Job Description : \n${jobDescription}.\n Projects : \n${project} \nSkills: \n${skills} \nExperience: \n${experience}`
+  );
+  const response = result.response;
+  return response.text();
+}