improve detector

src/generateFiles.mts

@@ -7,7 +7,9 @@ import { getStreamlitApp } from './getStreamlitApp.mts'
 import { getWebApp } from './getWebApp.mts'
 import { getReactApp } from './getReactApp.mts'
 import { isStreamlitAppPrompt } from './isStreamlitAppPrompt.mts'
-import { isPythonOrGradioAppPrompt } from './isPythonOrGradioAppPrompt.mts'
+import { isPythonAppPrompt } from './isPythonAppPrompt.mts'
+import { isGradioAppPrompt } from './isGradioAppPrompt.mts'
+import { isWebAppPrompt } from './isWebAppPrompt.mts'
 import { isReactAppPrompt } from './isReactAppPrompt.mts'
 
 export const generateFiles = async (
@@ -20,12 +22,16 @@ export const generateFiles = async (
   }
 
   const { prefix, files, instructions } =
-  isStreamlitAppPrompt(prompt)
+    isWebAppPrompt(prompt)
+    ? getWebApp(prompt)
+    : isStreamlitAppPrompt(prompt)
     ? getStreamlitApp(prompt)
-    : isPythonOrGradioAppPrompt(prompt)
+    : isGradioAppPrompt(prompt)
     ? getGradioApp(prompt)
     : isReactAppPrompt(prompt)
     ? getReactApp(prompt)
+    : isPythonAppPrompt(prompt)
+    ? getGradioApp(prompt)
     : getWebApp(prompt)
 
   const inputs = createLlamaCoderPrompt(instructions) + "\nSure! Here are the source files:\n" + prefix

src/isGradioAppPrompt.mts

@@ -0,0 +1,12 @@
+export function isGradioAppPrompt(prompt: string) {
+  const lowerCasePrompt = prompt.toLocaleLowerCase().trim()
+  return (
+    lowerCasePrompt.startsWith("code a gradio") ||
+    lowerCasePrompt.startsWith("make a gradio") ||
+    lowerCasePrompt.startsWith("build a gradio") ||
+    lowerCasePrompt.startsWith("create a gradio") ||
+    lowerCasePrompt.startsWith("a gradio") ||
+    lowerCasePrompt.startsWith("using gradio") ||
+    lowerCasePrompt.startsWith("with gradio")
+  )
+}

src/isPythonAppPrompt.mts

@@ -0,0 +1,11 @@
+export function isPythonAppPrompt(prompt: string) {
+  const lowerCasePrompt = prompt.toLocaleLowerCase().trim()
+  return (
+    lowerCasePrompt.startsWith("code a python app") ||
+    lowerCasePrompt.startsWith("make a python app") ||
+    lowerCasePrompt.startsWith("build a python app") ||
+    lowerCasePrompt.startsWith("create a python app") ||
+    lowerCasePrompt.startsWith("a python app") ||
+    lowerCasePrompt.startsWith("python app")
+  )
+}

src/isPythonOrGradioAppPrompt.mts

@@ -1,5 +0,0 @@
-export function isPythonOrGradioAppPrompt(prompt: string) {
-  const lowerCasePrompt = prompt.toLocaleLowerCase()
-  return lowerCasePrompt.includes("python")
-     || lowerCasePrompt.includes("gradio")
-}

src/isReactAppPrompt.mts

@@ -1,9 +1,26 @@
 export function isReactAppPrompt(prompt: string) {
-  const lowerCasePrompt = prompt.toLocaleLowerCase()
-  return lowerCasePrompt.includes("react")
-     || lowerCasePrompt.includes("create react app")
-     || lowerCasePrompt.includes("reactjs")
-     || lowerCasePrompt.includes("next app")
-     || lowerCasePrompt.includes("nextjs app")
-     || lowerCasePrompt.includes("nextjs")
+  const lowerCasePrompt = prompt.toLocaleLowerCase().trim()
+  return (
+    lowerCasePrompt.startsWith("code a react") ||
+    lowerCasePrompt.startsWith("make a react") ||
+    lowerCasePrompt.startsWith("build a react") ||
+    lowerCasePrompt.startsWith("create a react") ||
+    lowerCasePrompt.startsWith("a react") ||
+    lowerCasePrompt.startsWith("using react") ||
+    lowerCasePrompt.startsWith("with react") ||
+    lowerCasePrompt.startsWith("code a nextjs") ||
+    lowerCasePrompt.startsWith("make a nextjs") ||
+    lowerCasePrompt.startsWith("build a nextjs") ||
+    lowerCasePrompt.startsWith("create a nextjs") ||
+    lowerCasePrompt.startsWith("a nextjs") ||
+    lowerCasePrompt.startsWith("using nextjs") ||
+    lowerCasePrompt.startsWith("with nextjs") ||
+    lowerCasePrompt.startsWith("code a next js") ||
+    lowerCasePrompt.startsWith("make a next js") ||
+    lowerCasePrompt.startsWith("build a next js") ||
+    lowerCasePrompt.startsWith("create a next js") ||
+    lowerCasePrompt.startsWith("a next js") ||
+    lowerCasePrompt.startsWith("using next js") ||
+    lowerCasePrompt.startsWith("with next js")
+  )
 }

src/isStreamlitAppPrompt.mts

@@ -1,4 +1,12 @@
 export function isStreamlitAppPrompt(prompt: string) {
-  const lowerCasePrompt = prompt.toLocaleLowerCase()
-  return lowerCasePrompt.includes("streamlit")
+  const lowerCasePrompt = prompt.toLocaleLowerCase().trim()
+  return (
+    lowerCasePrompt.startsWith("code a streamlit") ||
+    lowerCasePrompt.startsWith("make a streamlit") ||
+    lowerCasePrompt.startsWith("build a streamlit") ||
+    lowerCasePrompt.startsWith("create a streamlit") ||
+    lowerCasePrompt.startsWith("a streamlit") ||
+    lowerCasePrompt.startsWith("using streamlit") ||
+    lowerCasePrompt.startsWith("with streamlit")
+  )
 }

src/isWebAppPrompt.mts

@@ -0,0 +1,14 @@
+export function isWebAppPrompt(prompt: string) {
+  const lowerCasePrompt = prompt.toLocaleLowerCase().trim()
+  return (
+    lowerCasePrompt.startsWith("code a web") ||
+    lowerCasePrompt.startsWith("code an app") ||
+    lowerCasePrompt.startsWith("code a app") ||
+    lowerCasePrompt.startsWith("make a web") ||
+    lowerCasePrompt.startsWith("make an app") ||
+    lowerCasePrompt.startsWith("build a web") ||
+    lowerCasePrompt.startsWith("build an app") ||
+    lowerCasePrompt.startsWith("create a web") ||
+    lowerCasePrompt.startsWith("create an app")
+  )
+}