Upload folder using huggingface_hub

app/api/create-ai-sandbox-v2/route.ts

@@ -1,4 +1,4 @@
-import { NextResponse } from 'next/server';
+import { NextRequest, NextResponse } from 'next/server';
 import { SandboxFactory } from '@/lib/sandbox/factory';
 // SandboxProvider type is used through SandboxFactory
 import type { SandboxState } from '@/types/sandbox';
@@ -12,9 +12,10 @@ declare global {
   var sandboxState: SandboxState;
 }
 
-export async function POST() {
+export async function POST(request: NextRequest) {
   try {
-    console.log('[create-ai-sandbox-v2] Creating sandbox...');
+    const { runtime = 'react' } = await request.json() || {};
+    console.log(`[create-ai-sandbox-v2] Creating sandbox for ${runtime}...`);
     
     // Clean up all existing sandboxes
     console.log('[create-ai-sandbox-v2] Cleaning up existing sandboxes...');
@@ -41,8 +42,13 @@ export async function POST() {
     const provider = SandboxFactory.create();
     const sandboxInfo = await provider.createSandbox();
     
-    console.log('[create-ai-sandbox-v2] Setting up Vite React app...');
-    await provider.setupViteApp();
+    if (runtime === 'python') {
+      console.log('[create-ai-sandbox-v2] Setting up Python environment...');
+      await provider.setupPythonEnv();
+    } else {
+      console.log('[create-ai-sandbox-v2] Setting up Vite React app...');
+      await provider.setupViteApp();
+    }
     
     // Register with sandbox manager
     sandboxManager.registerSandbox(sandboxInfo.sandboxId, provider);
@@ -75,7 +81,7 @@ export async function POST() {
       sandboxId: sandboxInfo.sandboxId,
       url: sandboxInfo.url,
       provider: sandboxInfo.provider,
-      message: 'Sandbox created and Vite React app initialized'
+      message: `Sandbox created and ${runtime} initialized`
     });
 
   } catch (error) {

app/api/submit-job/route.ts

@@ -4,6 +4,7 @@ import { getProviderForModel } from "@/lib/ai/provider-manager";
 import { extractFilesFromAI, sanitizeBranchName } from "@/lib/ai-utils";
 import { pushToGitHub } from "@/lib/github";
 import { appConfig } from "@/config/app.config";
+import { fetchArxivPaper } from "@/lib/arxiv";
 
 export async function GET() {
   return NextResponse.json({ message: "Submit job endpoint is active. Use POST to submit a paper." });
@@ -15,7 +16,7 @@ export async function POST(request: NextRequest) {
       arxivUrl, 
       paperContent, 
       repo, 
-      paperName,
+      paperName: providedPaperName,
       model = appConfig.ai.defaultModel 
     } = await request.json();
 
@@ -23,37 +24,41 @@ export async function POST(request: NextRequest) {
       return NextResponse.json({ error: "Target GitHub repository is required (e.g. 'owner/repo')" }, { status: 400 });
     }
 
-    if (!paperName) {
-      return NextResponse.json({ error: "Paper name is required for branch naming" }, { status: 400 });
-    }
-
     const githubToken = process.env.PERSONAL_ACCESS_TOKEN;
     if (!githubToken) {
       return NextResponse.json({ error: "GitHub PERSONAL_ACCESS_TOKEN not configured in environment" }, { status: 500 });
     }
 
     let finalPaperContent = paperContent || "";
+    let paperName = providedPaperName || "paper-to-code";
 
-    // 1. Get paper content from Arxiv if URL is provided
+    // 1. Get paper content from Arxiv API if URL or query is provided
     if (arxivUrl && !paperContent) {
-      console.log(`[submit-job] Scraping Arxiv: ${arxivUrl}`);
-      try {
-        const scrapeResponse = await fetch(`${process.env.NEXT_PUBLIC_APP_URL || 'http://localhost:3000'}/api/scrape-website`, {
-          method: 'POST',
-          headers: { 'Content-Type': 'application/json' },
-          body: JSON.stringify({ url: arxivUrl })
-        });
-        
-        if (scrapeResponse.ok) {
-          const scrapeData = await scrapeResponse.json();
-          if (scrapeData.success) {
-            finalPaperContent = scrapeData.data.content;
-          } else {
-            console.warn(`[submit-job] Scrape failed: ${scrapeData.error}`);
+      console.log(`[submit-job] Fetching Arxiv paper: ${arxivUrl}`);
+      const paper = await fetchArxivPaper(arxivUrl);
+      if (paper) {
+        finalPaperContent = `Title: ${paper.title}\n\nAuthors: ${paper.authors.join(', ')}\n\nSummary: ${paper.summary}`;
+        paperName = paper.title;
+        console.log(`[submit-job] Successfully fetched: ${paper.title}`);
+      } else {
+        console.warn(`[submit-job] Arxiv API failed for ${arxivUrl}, falling back to scraping`);
+        // Fallback to scrape if API fails
+        try {
+          const scrapeResponse = await fetch(`${process.env.NEXT_PUBLIC_APP_URL || 'http://localhost:3000'}/api/scrape-website`, {
+            method: 'POST',
+            headers: { 'Content-Type': 'application/json' },
+            body: JSON.stringify({ url: arxivUrl })
+          });
+          
+          if (scrapeResponse.ok) {
+            const scrapeData = await scrapeResponse.json();
+            if (scrapeData.success) {
+              finalPaperContent = scrapeData.data.content;
+            }
           }
+        } catch (error) {
+          console.error(`[submit-job] Error calling scrape API:`, error);
         }
-      } catch (error) {
-        console.error(`[submit-job] Error calling scrape API:`, error);
       }
     }
 

app/generation/page.tsx

@@ -61,6 +61,7 @@ interface ScrapeData {
 }
 
 function AISandboxPage() {
+  const [mode, setMode] = useState<'website' | 'paper'>('website');
   const [sandboxData, setSandboxData] = useState<SandboxData | null>(null);
   const [loading, setLoading] = useState(false);
   const [status, setStatus] = useState({ text: 'Not connected', active: false });
@@ -529,18 +530,20 @@ function AISandboxPage() {
 
   const sandboxCreationRef = useRef<boolean>(false);
   
-  const createSandbox = async (fromHomeScreen = false) => {
+  const createSandbox = async (fromHomeScreen = false, runtimeOverride?: string) => {
     // Prevent duplicate sandbox creation
     if (sandboxCreationRef.current) {
       console.log('[createSandbox] Sandbox creation already in progress, skipping...');
       return null;
     }
     
+    const runtime = runtimeOverride || (mode === 'paper' ? 'python' : 'react');
+    
     sandboxCreationRef.current = true;
-    console.log('[createSandbox] Starting sandbox creation...');
+    console.log(`[createSandbox] Starting sandbox creation for ${runtime}...`);
     setLoading(true);
     setShowLoadingBackground(true);
-    updateStatus('Creating sandbox...', false);
+    updateStatus(`Creating ${runtime} sandbox...`, false);
     setResponseArea([]);
     setScreenshotError(null);
     
@@ -548,7 +551,7 @@ function AISandboxPage() {
       const response = await fetch('/api/create-ai-sandbox-v2', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({})
+        body: JSON.stringify({ runtime })
       });
       
       const data = await response.json();
@@ -3282,7 +3285,33 @@ Focus on the key sections and content, making it clean and modern.`;
     <HeaderProvider>
       <div className="font-sans bg-background text-foreground h-screen flex flex-col">
       <div className="bg-white py-[15px] py-[8px] border-b border-border-faint flex items-center justify-between shadow-sm">
-        <HeaderBrandKit />
+        <div className="flex items-center gap-4">
+          <HeaderBrandKit />
+          <div className="bg-black-alpha-4 p-2 rounded-8 flex gap-2 ml-4">
+            <button 
+              onClick={() => {
+                if (mode !== 'website') {
+                  setMode('website');
+                  setSandboxData(null); // Reset sandbox when switching modes
+                }
+              }}
+              className={`px-12 py-4 rounded-6 text-xs font-medium transition-all ${mode === 'website' ? 'bg-white shadow-sm text-accent-black' : 'text-black-alpha-48 hover:text-black-alpha-72'}`}
+            >
+              Website2Code
+            </button>
+            <button 
+              onClick={() => {
+                if (mode !== 'paper') {
+                  setMode('paper');
+                  setSandboxData(null); // Reset sandbox when switching modes
+                }
+              }}
+              className={`px-12 py-4 rounded-6 text-xs font-medium transition-all ${mode === 'paper' ? 'bg-white shadow-sm text-accent-black' : 'text-black-alpha-48 hover:text-black-alpha-72'}`}
+            >
+              Paper2Code
+            </button>
+          </div>
+        </div>
         <div className="flex items-center gap-2">
           {/* Model Selector - Left side */}
           <select

app/page.tsx

@@ -38,6 +38,7 @@ interface SearchResult {
 }
 
 export default function HomePage() {
+  const [mode, setMode] = useState<'website' | 'paper'>('website');
   const [url, setUrl] = useState<string>("");
   const [selectedStyle, setSelectedStyle] = useState<string>("1");
   const [selectedModel, setSelectedModel] = useState<string>(appConfig.ai.defaultModel);
@@ -87,7 +88,34 @@ export default function HomePage() {
     const inputValue = url.trim();
 
     if (!inputValue) {
-      toast.error("Please enter a URL or search term");
+      toast.error(mode === 'website' ? "Please enter a URL or search term" : "Please enter an Arxiv link or paper query");
+      return;
+    }
+
+    if (mode === 'paper') {
+      toast.info("Transforming paper to code...");
+      try {
+        const response = await fetch('/api/submit-job', {
+          method: 'POST',
+          headers: { 'Content-Type': 'application/json' },
+          body: JSON.stringify({ 
+            arxivUrl: inputValue,
+            repo: 'AUXteam/Paper2Code-Jobs', // Default repo for paper jobs
+            paperName: inputValue.split('/').pop() || 'paper'
+          })
+        });
+        const data = await response.json();
+        if (data.success) {
+          toast.success("Job submitted! Code will be pushed to GitHub.");
+          if (data.githubUrl) {
+            window.open(data.githubUrl, '_blank');
+          }
+        } else {
+          toast.error(data.error || "Failed to submit job");
+        }
+      } catch (error) {
+        toast.error("An error occurred while submitting the paper");
+      }
       return;
     }
 
@@ -260,10 +288,29 @@ export default function HomePage() {
             <HomeHeroBackground />
 
             <div className="relative container px-16">
+              <div className="flex justify-center mb-12">
+                <div className="bg-black-alpha-4 p-4 rounded-12 flex gap-4">
+                  <button 
+                    onClick={() => setMode('website')}
+                    className={`px-16 py-8 rounded-8 text-label-medium transition-all ${mode === 'website' ? 'bg-white shadow-sm text-accent-black' : 'text-black-alpha-48 hover:text-black-alpha-72'}`}
+                  >
+                    Website2Code
+                  </button>
+                  <button 
+                    onClick={() => setMode('paper')}
+                    className={`px-16 py-8 rounded-8 text-label-medium transition-all ${mode === 'paper' ? 'bg-white shadow-sm text-accent-black' : 'text-black-alpha-48 hover:text-black-alpha-72'}`}
+                  >
+                    Paper2Code
+                  </button>
+                </div>
+              </div>
+
               <HomeHeroBadge />
               <HomeHeroTitle />
               <p className="text-center text-body-large">
-                Clone brand format or re-imagine any website, in seconds.
+                {mode === 'website' 
+                  ? "Clone brand format or re-imagine any website, in seconds." 
+                  : "Transform research papers into functional Python or React code."}
               </p>
               <Link
                 className="bg-black-alpha-4 hover:bg-black-alpha-6 rounded-6 px-8 lg:px-6 text-label-large h-30 lg:h-24 block mt-8 mx-auto w-max gap-4 transition-all"
@@ -378,7 +425,7 @@ export default function HomePage() {
                       )}
                       <input
                         className="flex-1 bg-transparent text-body-input text-accent-black placeholder:text-black-alpha-48 focus:outline-none focus:ring-0 focus:border-transparent"
-                        placeholder="Enter URL or search term..."
+                        placeholder={mode === 'website' ? "Enter URL or search term..." : "Enter Arxiv link or reference..."}
                         type="text"
                         value={url}
                         disabled={isSearching}
@@ -416,7 +463,7 @@ export default function HomePage() {
                       >
                         <HeroInputSubmitButton 
                           dirty={url.length > 0} 
-                          buttonText={isURL(url) ? 'Scrape Site' : 'Search'} 
+                          buttonText={mode === 'website' ? (isURL(url) ? 'Scrape Site' : 'Search') : 'Submit Paper'} 
                           disabled={isSearching}
                         />
                       </div>
@@ -424,9 +471,9 @@ export default function HomePage() {
                   )}
                 </div>
 
-                {/* Options Section - Only show when valid URL */}
+                {/* Options Section - Only show when valid URL and in website mode */}
                 <div className={`overflow-hidden transition-all duration-500 ease-in-out ${
-                  isValidUrl ? (extendBrandStyles ? 'max-h-[400px]' : 'max-h-[300px]') + ' opacity-100' : 'max-h-0 opacity-0'
+                  mode === 'website' && isValidUrl ? (extendBrandStyles ? 'max-h-[400px]' : 'max-h-[300px]') + ' opacity-100' : 'max-h-0 opacity-0'
                 }`}>
                   <div className="px-[28px] pt-0 pb-[28px]">
                     <div className="border-t border-gray-100 bg-white">

lib/arxiv.ts

@@ -0,0 +1,63 @@
+import axios from 'axios';
+import { XMLParser } from 'fast-xml-parser';
+
+export interface ArxivPaper {
+  id: string;
+  title: string;
+  summary: string;
+  authors: string[];
+  published: string;
+  updated: string;
+  link: string;
+  pdfUrl: string;
+}
+
+const parser = new XMLParser();
+
+/**
+ * Fetches paper metadata from Arxiv API.
+ * Supports both direct Arxiv IDs and search queries.
+ */
+export async function fetchArxivPaper(query: string): Promise<ArxivPaper | null> {
+  try {
+    let id = '';
+    
+    // Extract ID if it's a URL
+    if (query.includes('arxiv.org/abs/')) {
+      id = query.split('arxiv.org/abs/')[1].split('?')[0];
+    } else if (query.includes('arxiv.org/pdf/')) {
+      id = query.split('arxiv.org/pdf/')[1].replace('.pdf', '').split('?')[0];
+    } else if (/^\d{4}\.\d{4,5}$/.test(query)) {
+      id = query;
+    }
+
+    const apiUrl = id 
+      ? `http://export.arxiv.org/api/query?id_list=${id}`
+      : `http://export.arxiv.org/api/query?search_query=all:${encodeURIComponent(query)}&max_results=1`;
+
+    console.log(`[arxiv] Fetching: ${apiUrl}`);
+    const response = await axios.get(apiUrl);
+    const jsonObj = parser.parse(response.data);
+
+    const entry = jsonObj.feed?.entry;
+    if (!entry) return null;
+
+    const paperEntry = Array.isArray(entry) ? entry[0] : entry;
+    
+    return {
+      id: paperEntry.id,
+      title: paperEntry.title.trim().replace(/\n/g, ' '),
+      summary: paperEntry.summary.trim(),
+      authors: Array.isArray(paperEntry.author) 
+        ? paperEntry.author.map((a: any) => a.name)
+        : [paperEntry.author.name],
+      published: paperEntry.published,
+      updated: paperEntry.updated,
+      link: paperEntry.id,
+      pdfUrl: paperEntry.id.replace('abs', 'pdf') + '.pdf'
+    };
+  } catch (error) {
+    console.error('[arxiv] Error fetching paper:', error);
+    return null;
+  }
+}

lib/sandbox/providers/e2b-provider.ts

@@ -228,6 +228,47 @@ export class E2BProvider extends SandboxProvider {
     };
   }
 
+  async setupPythonEnv(): Promise<void> {
+    if (!this.sandbox) {
+      throw new Error('No active sandbox');
+    }
+
+    console.log('[E2BProvider] Setting up Python environment...');
+    
+    // Create directory structure
+    await this.sandbox.runCode(`
+import os
+os.makedirs('/home/user/app', exist_ok=True)
+print('✓ /home/user/app created')
+    `);
+
+    // Create a requirements.txt with common scientific libraries
+    const requirements = `
+numpy
+pandas
+matplotlib
+scipy
+scikit-learn
+requests
+    `.trim();
+
+    await this.writeFile('requirements.txt', requirements);
+    
+    // Create a main.py
+    const mainPy = `
+def main():
+    print("Hello from Paper2Code Python Environment!")
+
+if __name__ == "__main__":
+    main()
+    `.trim();
+
+    await this.writeFile('main.py', mainPy);
+
+    // Install requirements
+    await this.runCommand('pip install -r requirements.txt');
+  }
+
   async setupViteApp(): Promise<void> {
     if (!this.sandbox) {
       throw new Error('No active sandbox');

lib/sandbox/types.ts

@@ -58,6 +58,11 @@ export abstract class SandboxProvider {
     throw new Error('setupViteApp not implemented for this provider');
   }
   
+  async setupPythonEnv(): Promise<void> {
+    // Default implementation for setting up a Python environment
+    throw new Error('setupPythonEnv not implemented for this provider');
+  }
+
   async restartViteServer(): Promise<void> {
     // Default implementation for restarting Vite
     throw new Error('restartViteServer not implemented for this provider');

package.json

@@ -53,12 +53,14 @@
     "@vercel/sandbox": "^0.0.17",
     "ai": "^5.0.0",
     "autoprefixer": "^10.4.21",
+    "axios": "^1.13.5",
     "class-variance-authority": "^0.7.1",
     "classnames": "^2.5.1",
     "clsx": "^2.1.1",
     "copy-to-clipboard": "^3.3.3",
     "cors": "^2.8.5",
     "dotenv": "^17.2.1",
+    "fast-xml-parser": "^5.3.6",
     "framer-motion": "^12.23.12",
     "groq-sdk": "^0.29.0",
     "jotai": "^2.14.0",

pnpm-lock.yaml

@@ -128,6 +128,9 @@ importers:
       autoprefixer:
         specifier: ^10.4.21
         version: 10.4.21(postcss@8.5.6)
+      axios:
+        specifier: ^1.13.5
+        version: 1.13.5
       class-variance-authority:
         specifier: ^0.7.1
         version: 0.7.1
@@ -146,6 +149,9 @@ importers:
       dotenv:
         specifier: ^17.2.1
         version: 17.2.1
+      fast-xml-parser:
+        specifier: ^5.3.6
+        version: 5.3.6
       framer-motion:
         specifier: ^12.23.12
         version: 12.23.12(react-dom@19.1.0(react@19.1.0))(react@19.1.0)
@@ -1763,8 +1769,8 @@ packages:
     resolution: {integrity: sha512-Xm7bpRXnDSX2YE2YFfBk2FnF0ep6tmG7xPh8iHee8MIcrgq762Nkce856dYtJYLkuIoYZvGfTs/PbZhideTcEg==}
     engines: {node: '>=4'}
 
-  axios@1.11.0:
-    resolution: {integrity: sha512-1Lx3WLFQWm3ooKDYZD1eXmoGO9fxYQjrycfHFC8P0sCfQVXyROp0p9PFWBehewBOdCwHc+f/b8I0fMto5eSfwA==}
+  axios@1.13.5:
+    resolution: {integrity: sha512-cz4ur7Vb0xS4/KUN0tPWe44eqxrIu31me+fbang3ijiNscE129POzipJJA6zniq2C/Z6sJCjMimjS8Lc/GAs8Q==}
 
   axobject-query@4.1.0:
     resolution: {integrity: sha512-qIj0G9wZbMGNLjLmg1PT6v2mE9AH2zlnADJD/2tC6E00hgmhUOfEB6greHPAfLRSufHqROIUTkw6E+M3lH0PTQ==}
@@ -2195,6 +2201,10 @@ packages:
   fast-levenshtein@2.0.6:
     resolution: {integrity: sha512-DCXu6Ifhqcks7TZKY3Hxp3y6qphY5SJZmrWMDrKcERSOXWQdMhU9Ig/PYrzyw/ul9jOIyh0N4M0tbC5hodg8dw==}
 
+  fast-xml-parser@5.3.6:
+    resolution: {integrity: sha512-QNI3sAvSvaOiaMl8FYU4trnEzCwiRr8XMWgAHzlrWpTSj+QaCSvOf1h82OEP1s4hiAXhnbXSyFWCf4ldZzZRVA==}
+    hasBin: true
+
   fastq@1.19.1:
     resolution: {integrity: sha512-GwLTyxkCXjXbxqIhTsMI2Nui8huMPtnxg7krajPJAjnEG/iiOS7i+zCtWGZR9G0NBKbXKh6X9m9UIsYX/N6vvQ==}
 
@@ -2252,6 +2262,10 @@ packages:
     resolution: {integrity: sha512-KrGhL9Q4zjj0kiUt5OO4Mr/A/jlI2jDYs5eHBpYHPcBEVSiipAvn2Ko2HnPe20rmcuuvMHNdZFp+4IlGTMF0Ow==}
     engines: {node: '>= 6'}
 
+  form-data@4.0.5:
+    resolution: {integrity: sha512-8RipRLol37bNs2bhoV67fiTEvdTrbMUYcFTiy3+wuuOnUog2QBHCZWXDRijWQfAkhBj2Uf5UnVaiWwA5vdd82w==}
+    engines: {node: '>= 6'}
+
   format@0.2.2:
     resolution: {integrity: sha512-wzsgA6WOq+09wrU1tsJ09udeR/YZRaeArL9e1wPbFg3GG2yDnC2ldKpxs4xunpFF9DgqCqOIra3bc1HWrJ37Ww==}
     engines: {node: '>=0.4.x'}
@@ -3278,6 +3292,9 @@ packages:
     resolution: {integrity: sha512-6fPc+R4ihwqP6N/aIv2f1gMH8lOVtWQHoqC4yK6oSDVVocumAsfCqjkXnqiYMhmMwS/mEHLp7Vehlt3ql6lEig==}
     engines: {node: '>=8'}
 
+  strnum@2.1.2:
+    resolution: {integrity: sha512-l63NF9y/cLROq/yqKXSLtcMeeyOfnSQlfMSlzFt/K73oIaD8DGaQWd7Z34X9GPiKqP5rbSh84Hl4bOlLcjiSrQ==}
+
   styled-jsx@5.1.6:
     resolution: {integrity: sha512-qSVyDTeMotdvQYoHWLNGwRFJHC+i+ZvdBRYosOFgC+Wg1vx4frN2/RG/NA7SYqqvKNLf39P2LSRA2pu6n0XYZA==}
     engines: {node: '>= 12.0.0'}
@@ -3829,7 +3846,7 @@ snapshots:
 
   '@mendable/firecrawl-js@4.3.4':
     dependencies:
-      axios: 1.11.0
+      axios: 1.13.5
       typescript-event-target: 1.1.1
       zod: 3.25.76
       zod-to-json-schema: 3.24.6(zod@3.25.76)
@@ -5116,10 +5133,10 @@ snapshots:
 
   axe-core@4.10.3: {}
 
-  axios@1.11.0:
+  axios@1.13.5:
     dependencies:
       follow-redirects: 1.15.11
-      form-data: 4.0.4
+      form-data: 4.0.5
       proxy-from-env: 1.1.0
     transitivePeerDependencies:
       - debug
@@ -5692,6 +5709,10 @@ snapshots:
 
   fast-levenshtein@2.0.6: {}
 
+  fast-xml-parser@5.3.6:
+    dependencies:
+      strnum: 2.1.2
+
   fastq@1.19.1:
     dependencies:
       reusify: 1.1.0
@@ -5745,6 +5766,14 @@ snapshots:
       hasown: 2.0.2
       mime-types: 2.1.35
 
+  form-data@4.0.5:
+    dependencies:
+      asynckit: 0.4.0
+      combined-stream: 1.0.8
+      es-set-tostringtag: 2.1.0
+      hasown: 2.0.2
+      mime-types: 2.1.35
+
   format@0.2.2: {}
 
   formdata-node@4.4.1:
@@ -6836,6 +6865,8 @@ snapshots:
 
   strip-json-comments@3.1.1: {}
 
+  strnum@2.1.2: {}
+
   styled-jsx@5.1.6(react@19.1.0):
     dependencies:
       client-only: 0.0.1