Adding history matter

server/app.py

@@ -4,17 +4,13 @@ from pydantic import BaseModel
 from fastapi.staticfiles import StaticFiles
 from services.generate_questions_service import GenerateQuestionsService
 
-# from data.load_data import retriever_pre
-
 generate_questions_service = GenerateQuestionsService()
 
-
 class Body(BaseModel):
-    school_subject: str
+    matter: str
     subject: str
     difficultie: str
 
-
 app = FastAPI()
 
 app.add_middleware(
@@ -24,20 +20,18 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
 @app.post("/generate_questions")
 async def generate_questions(body: Body):
-    school_subject = body.school_subject
+    print(body)
+    matter = body.matter
     subject = body.subject
     difficultie = body.difficultie
-    query = f"Quero que você gere questões de {school_subject}, sendo do assunto: {subject} e sendo da dificuldade: {difficultie}."
+    query = f"Quero que você gere questões de {matter}, sendo do assunto: {subject} e sendo da dificuldade: {difficultie}."
     res = generate_questions_service.handle(f"""{query}""")
     return res
 
-
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
 if __name__ == "__main__":
     import uvicorn
-
     uvicorn.run("app:app", host="0.0.0.0", port=8000)

web/src/App.jsx

@@ -18,8 +18,15 @@ function App() {
   const [menuState, setMenuState] = useState(true);
   const [isLoading, setIsLoading] = useState(false);
   const [questions, setQuestions] = useState(null);
+  const [selectedMatter, setSelectedMatter] = useState("biologia");
 
-  const subjects = [
+
+  const matter = [
+    { label: "Biologia", value: "biologia" },
+    { label: "História", value: "historia" },
+  ];
+
+  const biologySubjects = [
     { label: "Fisiologia", value: "fisiologia" },
     { label: "Embriologia", value: "embriologia" },
     { label: "Citologia", value: "citologia" },
@@ -31,6 +38,15 @@ function App() {
     { label: "Biofísica", value: "biofisica" },
   ]
 
+  const historySubjects = [
+    { label: "Revoluções", value: "revoluções" },
+    { label: "Grécia", value: "grécia" },
+    { label: "Roma", value: "roma" },
+    { label: "Primeira Guerra Mundial", value: "primeira guerra mundial" },
+    { label: "Guerra Fria", value: "guerra fria" },
+    { label: "Feudalismo", value: "feudalismo" },
+  ]
+
   const difficulties = [
     { label: "Fácil", value: "facil" },
     { label: "Intermediária", value: "intermediaria" },
@@ -46,14 +62,13 @@ function App() {
           "Content-Type": "application/json"
         },
         body: JSON.stringify({
+          matter: selectedMatter,
           subject, difficultie
         })
       });
-      console.log(res)
       if (res.ok) {
         const data = await res.json();
-        console.log(data)
-        setQuestions(data.questions);
+        setQuestions(data.rag_result.questions);
       } else {
         toast.error("Erro ao gerar questões", {
           position: "top-right",
@@ -114,12 +129,16 @@ function App() {
               <ChevronLeft className="w-10 h-10" onClick={e => setMenuState(prev => !prev)} />
             </div>
             <h2>Selecione uma matéria:</h2>
-            <Select defaultValue="biologia" >
+            <Select defaultValue="biologia" onValueChange={value => setSelectedMatter(value)}>
               <SelectTrigger className="w-[180px]">
                 <SelectValue placeholder="Matéria" />
               </SelectTrigger>
               <SelectContent >
-                <SelectItem value="biologia">Biologia</SelectItem>
+                {
+                  matter.map(({ label, value }) => (
+                    <SelectItem key={value} value={value}>{label}</SelectItem>
+                  ))
+                }
               </SelectContent>
             </Select>
             <h2>Selecione um conteúdo:</h2>
@@ -128,9 +147,16 @@ function App() {
                 <SelectValue placeholder="Conteúdo" />
               </SelectTrigger>
               <SelectContent onChange={e => console.log(e)}>
-                {subjects && subjects.map(({ label, value }) => (
-                  <SelectItem key={value} value={value}>{label}</SelectItem>
-                ))}
+                {
+                  selectedMatter === "biologia" ?
+                    biologySubjects.map(({ label, value }) => (
+                      <SelectItem key={value} value={value}>{label}</SelectItem>
+                    ))
+                    :
+                    historySubjects.map(({ label, value }) => (
+                      <SelectItem key={value} value={value}>{label}</SelectItem>
+                    ))
+                }
               </SelectContent>
             </Select>
             <h2>Selecione uma dificuldade:</h2>