Ready for Hugging Face Spaces deployment

pickup-line-generator/.env.production

@@ -0,0 +1,3 @@
+NEXT_PUBLIC_HF_TOKEN=your_huggingface_token
+NEXT_PUBLIC_MODEL_ID=HuggingFaceTB/SmolLM-135M
+NODE_ENV=production 

pickup-line-generator/app.py

@@ -0,0 +1,19 @@
+from pathlib import Path
+import subprocess
+import sys
+
+def setup():
+    # Install dependencies
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
+    
+    # Install npm dependencies and build Next.js app
+    subprocess.check_call(["npm", "install"])
+    subprocess.check_call(["npm", "run", "build"])
+
+def main():
+    # Start the Next.js server
+    subprocess.check_call(["npm", "start"])
+
+if __name__ == "__main__":
+    setup()
+    main() 

pickup-line-generator/app/api/generate/route.ts

@@ -1,58 +1,29 @@
 import { NextResponse } from 'next/server';
-import { getVibeGuidance } from '@/lib/prompts';
 
 export async function POST(request: Request) {
-  // Handle CORS
-  if (request.method === 'OPTIONS') {
-    return new NextResponse(null, {
-      status: 200,
+  try {
+    const body = await request.json();
+    
+    // Forward the request to our model server
+    const response = await fetch(`${process.env.NEXT_PUBLIC_API_URL}/generate`, {
+      method: 'POST',
       headers: {
-        'Access-Control-Allow-Origin': '*',
-        'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, OPTIONS',
-        'Access-Control-Allow-Headers': 'Content-Type, Authorization',
+        'Content-Type': 'application/json',
       },
+      body: JSON.stringify(body),
     });
-  }
 
-  try {
-    const { vibe } = await request.json();
-    
-    // Get the vibe guidance
-    const vibeGuide = getVibeGuidance(vibe);
-    
-    // Create the prompt
-    const prompt = `Instructions: Generate a pickup line with a ${vibe} vibe.\n${vibeGuide}`;
-    
-    // TODO: Replace this with actual model inference
-    // For now, return a mock response
-    const mockResponses = {
-      romantic: "Are you a magician? Because whenever I look at you, everyone else disappears. ❤️",
-      cheesy: "Are you a parking ticket? Because you've got FINE written all over you! 😏",
-      nerdy: "Are you made of copper and tellurium? Because you're Cu-Te! 🔬",
-      cringe: "Are you a dictionary? Because you're adding meaning to my life! 📚",
-      flirty: "Is your name Google? Because you've got everything I've been searching for! 😏"
-    };
+    if (!response.ok) {
+      throw new Error('Failed to generate pickup line');
+    }
 
-    const response = NextResponse.json({ pickupLine: mockResponses[vibe as keyof typeof mockResponses] });
-    
-    // Add CORS headers to the response
-    response.headers.set('Access-Control-Allow-Origin', '*');
-    response.headers.set('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
-    response.headers.set('Access-Control-Allow-Headers', 'Content-Type, Authorization');
-    
-    return response;
+    const data = await response.json();
+    return NextResponse.json(data);
   } catch (error) {
     console.error('Error generating pickup line:', error);
-    const errorResponse = NextResponse.json(
+    return NextResponse.json(
       { error: 'Failed to generate pickup line' },
       { status: 500 }
     );
-    
-    // Add CORS headers to the error response
-    errorResponse.headers.set('Access-Control-Allow-Origin', '*');
-    errorResponse.headers.set('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
-    errorResponse.headers.set('Access-Control-Allow-Headers', 'Content-Type, Authorization');
-    
-    return errorResponse;
   }
 } 

pickup-line-generator/next.config.js

@@ -1,7 +1,11 @@
 /** @type {import('next').NextConfig} */
 const nextConfig = {
+  output: 'standalone',
   reactStrictMode: true,
   swcMinify: true,
+  experimental: {
+    serverActions: true,
+  },
 }
 
 module.exports = nextConfig 

pickup-line-generator/requirements.txt

@@ -0,0 +1,6 @@
+torch>=2.0.0
+transformers>=4.30.0
+huggingface-hub>=0.19.0
+numpy>=1.24.0
+tqdm>=4.65.0
+typing-extensions>=4.5.0