first commit

.gitignore

@@ -0,0 +1,33 @@
+# Cache directory (downloaded private repo)
+cache/
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.Python
+
+# Environment files
+.env
+.venv
+env/
+venv/
+
+# IDE files
+.vscode/
+.idea/
+*.swp
+*.swo
+
+# OS files
+.DS_Store
+Thumbs.db
+
+# Logs
+*.log
+logs/
+
+# Temporary files
+*.tmp
+*.temp

Dockerfile

@@ -0,0 +1,154 @@
+
+
+# # Use NVIDIA CUDA base image (Ubuntu 22.04 jammy)
+# FROM nvidia/cuda:12.3.2-devel-ubuntu22.04
+
+# # Non-interactive apt
+# ENV DEBIAN_FRONTEND=noninteractive
+# ENV PYTHONUNBUFFERED=1
+# ENV CUDA_HOME=/usr/local/cuda
+# ENV PATH=${CUDA_HOME}/bin:/usr/local/bin:${PATH}
+# ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+
+# # System deps + Deadsnakes PPA for Python 3.12
+# RUN apt-get update && apt-get install -y --no-install-recommends \
+#     software-properties-common \
+#     ca-certificates \
+#     curl wget git build-essential \
+#     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 \
+#     libgomp1 libgcc-s1 \
+#  && add-apt-repository ppa:deadsnakes/ppa -y \
+#  && apt-get update && apt-get install -y --no-install-recommends \
+#     python3.12 python3.12-dev python3.12-venv \
+#  && rm -rf /var/lib/apt/lists/*
+
+# # Install pip for Python 3.12
+# RUN curl -sS https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py \
+#  && python3.12 /tmp/get-pip.py \
+#  && rm /tmp/get-pip.py
+
+# # Make 'python' and 'pip' point to 3.12
+# RUN ln -sf /usr/bin/python3.12 /usr/bin/python && \
+#     ln -sf /usr/local/bin/pip3.12 /usr/local/bin/pip
+
+# # ---------------- Writable app data & HF cache ----------------
+# ENV APP_DATA_DIR=/data
+# ENV HF_HOME=/data/hf_home
+# ENV HUGGINGFACE_HUB_CACHE=/data/hf_home
+# ENV TRANSFORMERS_CACHE=/data/hf_home
+# ENV MPLCONFIGDIR=/data/matplotlib   
+
+# RUN mkdir -p /data/geometry /data/solution /data/weights /data/hf_home /data/matplotlib \
+#  && chmod -R 777 /data
+
+# # ---------------- Install frpc for Gradio share=True ----------------
+# RUN mkdir -p /data/hf_home/gradio/frpc && \
+#     wget https://cdn-media.huggingface.co/frpc-gradio-0.3/frpc_linux_amd64 -O /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3 && \
+#     chmod +x /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3
+
+# # ---------------- Application setup ----------------
+# WORKDIR /app
+
+# # Copy requirements first (better caching)
+# COPY requirements.txt .
+
+# # Install Python deps
+# RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
+#  && pip install --no-cache-dir -r requirements.txt
+
+# # Copy the rest of the source
+# COPY . .
+
+# # Create cache directory for downloaded repos
+# RUN mkdir -p /app/cache && chmod 777 /app/cache
+
+# # Permissions
+# RUN chmod +x app.py
+# RUN chown -R 1000:1000 /app
+
+# # Expose Gradio port
+# EXPOSE 7860
+
+# # Run app
+# CMD ["python", "app.py"]
+
+
+
+
+# Use NVIDIA CUDA base image (Ubuntu 22.04 jammy)
+FROM nvidia/cuda:12.3.2-devel-ubuntu22.04
+
+# Non-interactive apt
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV CUDA_HOME=/usr/local/cuda
+ENV PATH=${CUDA_HOME}/bin:/usr/local/bin:${PATH}
+ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
+
+# Proxy (auto-configurable: overridden at runtime if needed)
+ENV http_proxy=""
+ENV https_proxy=""
+ENV HTTP_PROXY=""
+ENV HTTPS_PROXY=""
+
+# System deps + Deadsnakes PPA for Python 3.12
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    software-properties-common \
+    ca-certificates \
+    curl wget git build-essential \
+    libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 \
+    libgomp1 libgcc-s1 \
+ && add-apt-repository ppa:deadsnakes/ppa -y \
+ && apt-get update && apt-get install -y --no-install-recommends \
+    python3.12 python3.12-dev python3.12-venv \
+ && rm -rf /var/lib/apt/lists/*
+
+# Install pip for Python 3.12
+RUN curl -sS https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py \
+ && python3.12 /tmp/get-pip.py \
+ && rm /tmp/get-pip.py
+
+# Make 'python' and 'pip' point to 3.12
+RUN ln -sf /usr/bin/python3.12 /usr/bin/python && \
+    ln -sf /usr/local/bin/pip3.12 /usr/local/bin/pip
+
+# ---------------- Writable app data & HF cache ----------------
+ENV APP_DATA_DIR=/data
+ENV HF_HOME=/data/hf_home
+ENV HUGGINGFACE_HUB_CACHE=/data/hf_home
+ENV TRANSFORMERS_CACHE=/data/hf_home
+ENV MPLCONFIGDIR=/data/matplotlib   
+
+RUN mkdir -p /data/geometry /data/solution /data/weights /data/hf_home /data/matplotlib \
+ && chmod -R 777 /data
+
+# ---------------- Install frpc for Gradio share=True ----------------
+RUN mkdir -p /data/hf_home/gradio/frpc && \
+    wget https://cdn-media.huggingface.co/frpc-gradio-0.3/frpc_linux_amd64 -O /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3 && \
+    chmod +x /data/hf_home/gradio/frpc/frpc_linux_amd64_v0.3
+
+# ---------------- Application setup ----------------
+WORKDIR /app
+
+# Copy requirements first (better caching)
+COPY requirements.txt .
+
+# Install Python deps
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
+ && pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the source
+COPY . .
+
+# Create cache directory for downloaded repos
+RUN mkdir -p /app/cache && chmod 777 /app/cache
+
+# Permissions
+RUN chmod +x app.py
+RUN chown -R 1000:1000 /app
+
+# Expose Gradio port
+EXPOSE 7860
+
+# Run app
+CMD ["python", "app.py"]

app.py

@@ -0,0 +1,103 @@
+
+
+import os
+import sys
+import tempfile
+from pathlib import Path
+import traceback
+
+# Proxy support (office networks often require this)
+proxy = os.environ.get("HTTP_PROXY") or os.environ.get("http_proxy")
+if proxy:
+    print(f"🌐 Using proxy: {proxy}")
+else:
+    print("🚀 No proxy detected, connecting directly")
+
+# Writable base dir
+DATA_DIR = os.environ.get("APP_DATA_DIR", os.path.join(tempfile.gettempdir(), "appdata"))
+os.makedirs(DATA_DIR, exist_ok=True)
+os.environ.setdefault("MPLCONFIGDIR", DATA_DIR)
+
+def setup_environment():
+    cache_dir = Path("cache")
+    cache_dir.mkdir(exist_ok=True)
+    if str(cache_dir.absolute()) not in sys.path:
+        sys.path.insert(0, str(cache_dir.absolute()))
+    return cache_dir
+
+def download_private_repo(cache_dir, repo_id, hf_token):
+    from huggingface_hub import snapshot_download
+    import shutil
+    import os
+    print(f"🔍 Downloading repo {repo_id}")
+    repo_path = snapshot_download(
+        repo_id=repo_id,
+        token=hf_token,
+        repo_type="space",
+        local_dir=cache_dir,
+        local_dir_use_symlinks=False,
+    )
+    
+    # Copy everything from the repository to current working directory
+    print(f"📁 Copying all files and directories from {repo_path} to current directory")
+    
+    # Get all items in the repository
+    for item in os.listdir(repo_path):
+        src_path = os.path.join(repo_path, item)
+        dst_path = item
+        
+        # Skip hidden files and directories (like .git, .gitignore, etc.)
+        if item.startswith('.'):
+            print(f"⏭️ Skipping hidden item: {item}")
+            continue
+            
+        if os.path.isdir(src_path):
+            print(f"📁 Copying directory: {item}")
+            if os.path.exists(dst_path):
+                shutil.rmtree(dst_path)
+            shutil.copytree(src_path, dst_path)
+            print(f"✅ Directory {item} copied successfully")
+        elif os.path.isfile(src_path):
+            print(f"📄 Copying file: {item}")
+            shutil.copy2(src_path, dst_path)
+            print(f"✅ File {item} copied successfully")
+    
+    print(f"🎉 All repository contents copied to current working directory")
+    
+    return repo_path
+
+def load_private_app(repo_path):
+    sys.path.insert(0, str(repo_path))
+    import app as private_app
+    if hasattr(private_app, "demo"):
+        return private_app.demo
+    if hasattr(private_app, "app"):
+        return private_app.app
+    raise RuntimeError("No demo/app found in private repo")
+
+def main():
+    hf_token = os.environ.get("HF_TOKEN")
+    repo_id = os.environ.get("REPO_ID")
+
+    if not hf_token or not repo_id:
+        raise ValueError("HF_TOKEN and REPO_ID must be set")
+
+    print(f"🚀 Starting app with repo_id: {repo_id}")
+    print(f"📂 Current working directory: {os.getcwd()}")
+    
+    cache_dir = setup_environment()
+    repo_path = download_private_repo(cache_dir, repo_id, hf_token)    
+    demo = load_private_app(repo_path)
+
+    
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_error=True,
+        allowed_paths=[str(repo_path), DATA_DIR]
+    )
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,40 @@
+
+huggingface_hub>=0.16.0
+--index-url https://download.pytorch.org/whl/cu121
+--extra-index-url https://pypi.org/simple
+
+torch==2.5.0
+torchvision
+torchaudio
+
+torch_geometric
+
+-f https://data.pyg.org/whl/torch-2.5.0+cu121.html
+pyg_lib
+torch_scatter
+torch_sparse
+torch_cluster
+torch_spline_conv
+
+matplotlib
+vtk
+timm
+einops
+seaborn
+pyvista
+imageio
+pandas
+meshio
+accelerate
+omegaconf
+tensorboard
+lion-pytorch
+scikit-learn
+
+gradio==5.44.1
+pathlib2>=2.3.0
+typing-extensions>=4.0.0
+transformers>=4.20.0
+trimesh
+gradio_client==1.12.1
+