update for streamlit deploy

app.py

@@ -1358,19 +1358,74 @@ with gr.Blocks(
         # Save code to a temporary file
         if sdk == "static":
             file_name = "index.html"
+            with tempfile.NamedTemporaryFile("w", suffix=".html", delete=False) as f:
+                f.write(code)
+                temp_path = f.name
+            files_to_upload = [(temp_path, file_name)]
+        elif sdk == "docker":  # Streamlit (Python)
+            import shutil
+            import os
+            import tempfile
+            # Create a temp dir for the structure
+            temp_dir = tempfile.mkdtemp()
+            # 1. Write requirements.txt
+            reqs = "altair\npandas\nstreamlit\n"
+            req_path = os.path.join(temp_dir, "requirements.txt")
+            with open(req_path, "w") as f:
+                f.write(reqs)
+            # 2. Write Dockerfile
+            dockerfile_content = '''FROM python:3.9-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt ./
+COPY src/ ./src/
+
+RUN pip3 install -r requirements.txt
+
+EXPOSE 8501
+
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+'''
+            dockerfile_path = os.path.join(temp_dir, "Dockerfile")
+            with open(dockerfile_path, "w") as f:
+                f.write(dockerfile_content)
+            # 3. Write src/streamlit_app.py
+            src_dir = os.path.join(temp_dir, "src")
+            os.makedirs(src_dir, exist_ok=True)
+            app_py_path = os.path.join(src_dir, "streamlit_app.py")
+            with open(app_py_path, "w") as f:
+                f.write(code)
+            # Prepare files to upload
+            files_to_upload = [
+                (req_path, "requirements.txt"),
+                (dockerfile_path, "Dockerfile"),
+                (app_py_path, "src/streamlit_app.py"),
+            ]
         else:
             file_name = "app.py"
-        with tempfile.NamedTemporaryFile("w", suffix=f".{file_name.split('.')[-1]}", delete=False) as f:
-            f.write(code)
-            temp_path = f.name
-        # Upload the file
+            with tempfile.NamedTemporaryFile("w", suffix=".py", delete=False) as f:
+                f.write(code)
+                temp_path = f.name
+            files_to_upload = [(temp_path, file_name)]
+        # Upload all files
         try:
-            api.upload_file(
-                path_or_fileobj=temp_path,
-                path_in_repo=file_name,
-                repo_id=repo_id,
-                repo_type="space"
-            )
+            for local_path, repo_path in files_to_upload:
+                api.upload_file(
+                    path_or_fileobj=local_path,
+                    path_in_repo=repo_path,
+                    repo_id=repo_id,
+                    repo_type="space"
+                )
             space_url = f"https://huggingface.co/spaces/{repo_id}"
             return gr.update(value=f"✅ Deployed! [Open your Space here]({space_url})", visible=True)
         except Exception as e: