Merge pull request #1699 from Genez-io/cristim67/improv-streamlit-detect

feat: add pyproject.toml support for Streamlit detection

src/commands/analyze/frameworks.ts

@@ -461,10 +461,12 @@ export function isPythonLambdaFunction(contents: Record<string, string>): boolea
     return contents["requirements.txt"] !== undefined || contents["pyproject.toml"] !== undefined;
 }
 
-// Checks if the project is a Streamlit component (presence of 'requirements.txt' and 'streamlit' in 'requirements.txt')
+// Checks if the project is a Streamlit component (presence of 'requirements.txt' or 'pyproject.toml' and 'streamlit' in 'requirements.txt' or 'pyproject.toml')
 export function isStreamlitComponent(contents: Record<string, string>): boolean {
     return (
-        contents["requirements.txt"] !== undefined &&
-        contents["requirements.txt"].includes("streamlit")
+        (contents["requirements.txt"] !== undefined &&
+            contents["requirements.txt"].includes("streamlit")) ||
+        (contents["pyproject.toml"] !== undefined &&
+            contents["pyproject.toml"].includes("streamlit"))
     );
 }

src/commands/deploy/command.ts

@@ -209,12 +209,23 @@ async function decideDeployType(options: GenezioDeployOptions): Promise<DeployTy
         }
     }
 
-    // Check if requirements.txt exists
-    if (fs.existsSync(path.join(cwd, "requirements.txt"))) {
-        // Check if streamlit is in the requirements.txt
-        const requirementsTxt = fs.readFileSync(path.join(cwd, "requirements.txt"), "utf-8");
-        if (requirementsTxt.includes("streamlit")) {
-            return [DeployType.Streamlit];
+    // Check if requirements.txt or pyproject.toml exists
+    if (
+        fs.existsSync(path.join(cwd, "requirements.txt")) ||
+        fs.existsSync(path.join(cwd, "pyproject.toml"))
+    ) {
+        if (fs.existsSync(path.join(cwd, "requirements.txt"))) {
+            const requirementsTxt = fs.readFileSync(path.join(cwd, "requirements.txt"), "utf-8");
+            if (requirementsTxt.includes("streamlit")) {
+                return [DeployType.Streamlit];
+            }
+        }
+
+        if (fs.existsSync(path.join(cwd, "pyproject.toml"))) {
+            const pyprojectToml = fs.readFileSync(path.join(cwd, "pyproject.toml"), "utf-8");
+            if (pyprojectToml.includes("streamlit")) {
+                return [DeployType.Streamlit];
+            }
         }
     }