title generation prompt updated

app/api/generate/title/route.ts

@@ -13,10 +13,12 @@ export async function POST(
       ['x-use-cache']: "0"
     },
     body: JSON.stringify({
-      inputs: `Generate and return only a music title based on the following prompt: ${prompt}`,
+      // inputs: `Generate and return only a music title based on the following prompt: ${prompt}`,
+      inputs: `Generate a music title based on the following prompt: ${prompt}, return only the title between quotes.`,
       parameters: {
         num_return_sequences: 1,
-        return_full_text: false
+        return_full_text: false,
+
       }
     }),
   })

components/form.tsx

@@ -196,13 +196,13 @@ export const Form = ({ children }: { children: React.ReactNode }) => {
               ) : (
                 <div className="w-[300px] h-[300px] bg-stone-900 animate-pulse rounded-xl mx-auto"></div>
               )}
-              {/* {results.title ? (
+              {results.title ? (
                 <p className="text-center text-white font-bold text-3xl">
                   {results.title}
                 </p>
               ) : (
                 <div className="w-[450px] h-8 bg-stone-900 animate-pulse rounded-xl mx-auto"></div>
-              )} */}
+              )}
               {modelLoaded &&
                 (track !== "" ? (
                   <div className="mx-auto">

components/hooks/useGeneration.ts

@@ -54,21 +54,21 @@ export const useGeneration = () => {
       })
     })
 
-    // fetch("/api/generate/title", {
-    //   method: "POST",
-    //   headers: {
-    //     "Content-Type": "application/json"
-    //   },
-    //   body: JSON.stringify({
-    //     prompt: form.prompt
-    //   })
-    // }).then((res) => res.json())
-    // .then((res: any) => {
-    //   new_results.title = res.title
-    //   setResults({
-    //     ...new_results,
-    //   })
-    // })
+    fetch("/api/generate/title", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json"
+      },
+      body: JSON.stringify({
+        prompt: form.prompt
+      })
+    }).then((res) => res.json())
+    .then((res: any) => {
+      new_results.title = res.title
+      setResults({
+        ...new_results,
+      })
+    })
   }
 
   const randomize = () => {