Initial commit

.gitignore

@@ -0,0 +1,64 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# Virtual Environment
+venv/
+env/
+ENV/
+.env
+
+# IDE
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Model files and cache
+.cache/
+models/
+*.bin
+*.pt
+*.pth
+*.onnx
+*.h5
+*.hdf5
+
+# Logs
+*.log
+logs/
+
+# Gradio
+flagged/
+
+# Next.js
+.next/
+out/
+build/
+dist/
+.env*.local
+.env.development.local
+.env.test.local
+.env.production.local 

pickup-line-generator/.gitignore

@@ -0,0 +1,88 @@
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+
+# testing
+/coverage
+
+# next.js
+/.next/
+/out/
+
+# production
+/build
+
+# misc
+.DS_Store
+*.pem
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+
+# local env files
+.env*.local
+.env
+
+# vercel
+.vercel
+
+# typescript
+*.tsbuildinfo
+next-env.d.ts
+
+# IDE specific files
+.idea
+.vscode
+*.swp
+*.swo
+
+# OS generated files
+.DS_Store
+.DS_Store?
+._*
+.Spotlight-V100
+.Trashes
+ehthumbs.db
+Thumbs.db
+
+# Python virtual environment (if you're using Python for the model)
+venv/
+env/
+ENV/
+
+# Model files and cache
+.cache/
+models/
+*.bin
+*.pt
+*.pth
+*.onnx
+*.h5
+*.hdf5
+
+# Logs
+logs
+*.log
+
+# Optional npm cache directory
+.npm
+
+# Optional eslint cache
+.eslintcache
+
+# Optional REPL history
+.node_repl_history
+
+# Output of 'npm pack'
+*.tgz
+
+# Yarn Integrity file
+.yarn-integrity
+
+# dotenv environment variable files
+.env.development.local
+.env.test.local
+.env.production.local 

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

@@ -0,0 +1,58 @@
+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,
+      headers: {
+        'Access-Control-Allow-Origin': '*',
+        'Access-Control-Allow-Methods': 'GET, POST, PUT, DELETE, OPTIONS',
+        'Access-Control-Allow-Headers': 'Content-Type, Authorization',
+      },
+    });
+  }
+
+  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! 😏"
+    };
+
+    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;
+  } catch (error) {
+    console.error('Error generating pickup line:', error);
+    const errorResponse = 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/app/globals.css

@@ -0,0 +1,14 @@
+@tailwind base;
+@tailwind components;
+@tailwind utilities;
+
+:root {
+  --foreground-rgb: 0, 0, 0;
+  --background-start-rgb: 254, 246, 249;
+  --background-end-rgb: 255, 255, 255;
+}
+
+body {
+  color: rgb(var(--foreground-rgb));
+  background: rgb(var(--background-start-rgb));
+} 

pickup-line-generator/app/layout.tsx

@@ -0,0 +1,31 @@
+import type { Metadata } from 'next';
+import { Inter, Lobster } from 'next/font/google';
+import { Toaster } from 'react-hot-toast';
+import './globals.css';
+
+const inter = Inter({ subsets: ['latin'], variable: '--font-inter' });
+const lobster = Lobster({ 
+  weight: '400',
+  subsets: ['latin'],
+  variable: '--font-lobster',
+});
+
+export const metadata: Metadata = {
+  title: 'Pickup Line Generator',
+  description: 'Generate fun, clever, or cringey pickup lines using SmolLM!',
+};
+
+export default function RootLayout({
+  children,
+}: {
+  children: React.ReactNode;
+}) {
+  return (
+    <html lang="en">
+      <body className={`${inter.variable} ${lobster.variable} font-sans`}>
+        {children}
+        <Toaster position="bottom-center" />
+      </body>
+    </html>
+  );
+} 

pickup-line-generator/app/page.tsx

@@ -0,0 +1,120 @@
+'use client';
+
+import { useState } from 'react';
+import { toast } from 'react-hot-toast';
+
+const vibes = [
+  { id: 'romantic', label: '💝 Romantic', emoji: '❤️' },
+  { id: 'cheesy', label: '🧀 Cheesy', emoji: '😏' },
+  { id: 'nerdy', label: '🤓 Nerdy', emoji: '🔬' },
+  { id: 'cringe', label: '😬 Cringe', emoji: '😂' },
+  { id: 'flirty', label: '😘 Flirty', emoji: '💋' },
+];
+
+export default function Home() {
+  const [selectedVibe, setSelectedVibe] = useState(vibes[0].id);
+  const [pickupLine, setPickupLine] = useState('');
+  const [isLoading, setIsLoading] = useState(false);
+
+  const generatePickupLine = async () => {
+    setIsLoading(true);
+    try {
+      const response = await fetch('/api/generate', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({ vibe: selectedVibe }),
+      });
+      
+      if (!response.ok) throw new Error('Failed to generate pickup line');
+      
+      const data = await response.json();
+      setPickupLine(data.pickupLine);
+    } catch (error) {
+      toast.error('Failed to generate pickup line');
+      console.error(error);
+    } finally {
+      setIsLoading(false);
+    }
+  };
+
+  const copyToClipboard = async () => {
+    try {
+      await navigator.clipboard.writeText(pickupLine);
+      toast.success('Copied to clipboard!');
+    } catch (error) {
+      toast.error('Failed to copy');
+    }
+  };
+
+  return (
+    <main className="min-h-screen bg-pink-50 py-12 px-4 sm:px-6 lg:px-8">
+      <div className="max-w-2xl mx-auto">
+        <div className="text-center mb-12">
+          <h1 className="text-4xl font-display font-bold text-pink-600 mb-4">
+            💘 Pickup Line Generator
+          </h1>
+          <p className="text-gray-600">
+            Generate fun, clever, or cringey pickup lines using SmolLM! Select a vibe and click generate to get started! 😏
+          </p>
+        </div>
+
+        <div className="bg-white rounded-2xl shadow-xl p-8 space-y-6">
+          <div className="space-y-4">
+            <label htmlFor="vibe" className="block text-sm font-medium text-gray-700">
+              Choose a vibe
+            </label>
+            <select
+              id="vibe"
+              value={selectedVibe}
+              onChange={(e) => setSelectedVibe(e.target.value)}
+              className="mt-1 block w-full pl-3 pr-10 py-2 text-base border-gray-300 focus:outline-none focus:ring-pink-500 focus:border-pink-500 sm:text-sm rounded-md"
+            >
+              {vibes.map((vibe) => (
+                <option key={vibe.id} value={vibe.id}>
+                  {vibe.label}
+                </option>
+              ))}
+            </select>
+          </div>
+
+          <button
+            onClick={generatePickupLine}
+            disabled={isLoading}
+            className="w-full bg-gradient-to-r from-pink-500 to-pink-600 text-white py-3 px-4 rounded-lg font-medium hover:from-pink-600 hover:to-pink-700 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-pink-500 transition-all duration-200 shadow-md hover:shadow-lg disabled:opacity-50 disabled:cursor-not-allowed"
+          >
+            {isLoading ? 'Generating...' : 'Generate Line'}
+          </button>
+
+          {pickupLine && (
+            <div className="space-y-4">
+              <div className="bg-pink-50 rounded-lg p-4">
+                <p className="text-gray-800 text-lg">{pickupLine}</p>
+              </div>
+              
+              <div className="flex space-x-4">
+                <button
+                  onClick={generatePickupLine}
+                  className="flex-1 bg-white border border-pink-500 text-pink-500 py-2 px-4 rounded-lg font-medium hover:bg-pink-50 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-pink-500 transition-all duration-200"
+                >
+                  🔁 Generate Again
+                </button>
+                <button
+                  onClick={copyToClipboard}
+                  className="flex-1 bg-white border border-pink-500 text-pink-500 py-2 px-4 rounded-lg font-medium hover:bg-pink-50 focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-pink-500 transition-all duration-200"
+                >
+                  📋 Copy
+                </button>
+              </div>
+            </div>
+          )}
+        </div>
+
+        <div className="mt-8 text-center text-gray-500">
+          Built by Nath with SmolLM 🔥
+        </div>
+      </div>
+    </main>
+  );
+} 

pickup-line-generator/next.config.js

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

pickup-line-generator/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "pickup-line-generator",
+  "version": "0.1.0",
+  "private": true,
+  "scripts": {
+    "dev": "next dev",
+    "build": "next build",
+    "start": "next start",
+    "lint": "next lint"
+  },
+  "dependencies": {
+    "next": "14.1.0",
+    "react": "^18.2.0",
+    "react-dom": "^18.2.0",
+    "react-hot-toast": "^2.4.1"
+  },
+  "devDependencies": {
+    "@types/node": "20.17.27",
+    "@types/react": "18.3.20",
+    "@types/react-dom": "^18.2.19",
+    "autoprefixer": "^10.4.17",
+    "eslint": "^8.56.0",
+    "eslint-config-next": "14.1.0",
+    "postcss": "^8.4.35",
+    "tailwindcss": "^3.4.1",
+    "typescript": "5.8.2"
+  }
+}

pickup-line-generator/postcss.config.js

@@ -0,0 +1,6 @@
+module.exports = {
+  plugins: {
+    tailwindcss: {},
+    autoprefixer: {},
+  },
+} 

pickup-line-generator/tailwind.config.ts

@@ -0,0 +1,28 @@
+import type { Config } from 'tailwindcss';
+
+const config: Config = {
+  content: [
+    './pages/**/*.{js,ts,jsx,tsx,mdx}',
+    './components/**/*.{js,ts,jsx,tsx,mdx}',
+    './app/**/*.{js,ts,jsx,tsx,mdx}',
+  ],
+  theme: {
+    extend: {
+      fontFamily: {
+        sans: ['var(--font-inter)'],
+        display: ['var(--font-lobster)'],
+      },
+      colors: {
+        pink: {
+          50: '#fef6f9',
+          500: '#ff69b4',
+          600: '#ff1493',
+          700: '#db2777',
+        },
+      },
+    },
+  },
+  plugins: [],
+};
+
+export default config; 

pickup-line-generator/tsconfig.json

@@ -0,0 +1,27 @@
+{
+  "compilerOptions": {
+    "target": "es5",
+    "lib": ["dom", "dom.iterable", "esnext"],
+    "allowJs": true,
+    "skipLibCheck": true,
+    "strict": true,
+    "noEmit": true,
+    "esModuleInterop": true,
+    "module": "esnext",
+    "moduleResolution": "bundler",
+    "resolveJsonModule": true,
+    "isolatedModules": true,
+    "jsx": "preserve",
+    "incremental": true,
+    "plugins": [
+      {
+        "name": "next"
+      }
+    ],
+    "paths": {
+      "@/*": ["./*"]
+    }
+  },
+  "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", ".next/types/**/*.ts"],
+  "exclude": ["node_modules"]
+} 

pickup-line-generator/vercel.json

@@ -0,0 +1,18 @@
+{
+  "version": 2,
+  "builds": [
+    {
+      "src": "package.json",
+      "use": "@vercel/next"
+    }
+  ],
+  "routes": [
+    {
+      "src": "/(.*)",
+      "dest": "/"
+    }
+  ],
+  "env": {
+    "NEXT_PUBLIC_API_URL": "https://your-vercel-domain.vercel.app/api"
+  }
+} 

smollm_web_pickuplinegenerator.py

@@ -0,0 +1,216 @@
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+import re
+
+# Set page title and description
+title = "💖 Pickup Line Generator"
+description = """
+<div style="text-align: center; max-width: 650px; margin: 0 auto;">
+  <div>
+    <p>Generate fun, clever, or cringey pickup lines using SmolLM-135M! Select a vibe and click generate to get started! 😏</p>
+  </div>
+</div>
+"""
+
+# Load model and tokenizer
+print("Loading SmolLM-135M model...")
+MODEL_NAME = "HuggingFaceTB/SmolLM-135M"
+
+# Check for CUDA availability
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
+# Load the model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+# Set pad_token to eos_token to handle padding
+tokenizer.pad_token = tokenizer.eos_token
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME).to(device)
+
+print(f"Model loaded successfully! Memory footprint: {model.get_memory_footprint() / 1e6:.2f} MB")
+
+def get_vibe_guidance(vibe):
+    """Get specific guidance for each vibe with examples"""
+    vibe_patterns = {
+        "romantic": """Generate a romantic and sweet pickup line that's genuine and heartfelt.
+Example: 
+Input: Generate a romantic pickup line
+Output: Are you a magician? Because whenever I look at you, everyone else disappears. ❤️
+
+Now generate a romantic pickup line: """,
+
+        "cheesy": """Generate a super cheesy and over-the-top pickup line.
+Example:
+Input: Generate a cheesy pickup line
+Output: Are you a parking ticket? Because you've got FINE written all over you! 😏
+
+Now generate a cheesy pickup line: """,
+
+        "nerdy": """Generate a nerdy, science-themed pickup line.
+Example:
+Input: Generate a nerdy pickup line
+Output: Are you made of copper and tellurium? Because you're Cu-Te! 🔬
+
+Now generate a nerdy pickup line: """,
+
+        "cringe": """Generate the most cringey and over-the-top pickup line imaginable.
+Example:
+Input: Generate a cringe pickup line
+Output: Are you a dictionary? Because you're adding meaning to my life! 📚
+
+Now generate a cringe pickup line: """,
+
+        "flirty": """Generate a bold and flirty pickup line.
+Example:
+Input: Generate a flirty pickup line
+Output: Is your name Google? Because you've got everything I've been searching for! 😏
+
+Now generate a flirty pickup line: """
+    }
+    return vibe_patterns.get(vibe, "Generate a pickup line with a ")
+
+def generate_pickup_line(vibe):
+    """Generate a pickup line based on the selected vibe"""
+    # Get the vibe guidance
+    vibe_guide = get_vibe_guidance(vibe)
+    
+    # Create the prompt
+    prompt = f"""Instructions: Generate a pickup line with a {vibe} vibe.
+{vibe_guide}"""
+    
+    # Prepare inputs with explicit attention mask
+    encoded_input = tokenizer.encode_plus(
+        prompt,
+        return_tensors="pt",
+        padding=True,
+        return_attention_mask=True
+    )
+    input_ids = encoded_input["input_ids"].to(device)
+    attention_mask = encoded_input["attention_mask"].to(device)
+    
+    # Generate multiple responses and pick the best one
+    num_tries = 3
+    best_response = None
+    
+    for _ in range(num_tries):
+        with torch.no_grad():
+            outputs = model.generate(
+                input_ids,
+                attention_mask=attention_mask,
+                max_new_tokens=100,
+                do_sample=True,
+                temperature=0.8,
+                top_p=0.92,
+                top_k=50,
+                pad_token_id=tokenizer.eos_token_id,
+                eos_token_id=tokenizer.eos_token_id,
+            )
+        
+        # Get the full generated text
+        full_response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        
+        # Extract just the pickup line
+        if full_response.startswith(prompt):
+            response = full_response[len(prompt):].strip()
+        else:
+            response = full_response.replace(prompt, "").strip()
+        
+        # Clean up the response
+        for marker in ["Instructions:", "Generate a pickup line:", "\n"]:
+            if marker in response:
+                response = response.split(marker, 1)[0].strip()
+        
+        # Add appropriate emoji based on vibe
+        if vibe == "romantic":
+            response += " ❤️"
+        elif vibe == "cheesy":
+            response += " 😏"
+        elif vibe == "nerdy":
+            response += " 🔬"
+        elif vibe == "cringe":
+            response += " 😂"
+        elif vibe == "flirty":
+            response += " 💋"
+        
+        best_response = response
+        break
+    
+    return best_response
+
+# Create custom CSS
+custom_css = """
+.gradio-container {
+    background-color: #fef6f9 !important;
+}
+.title {
+    font-family: 'Lobster', cursive !important;
+    color: #ff69b4 !important;
+}
+.button {
+    background: linear-gradient(45deg, #ff69b4, #ff1493) !important;
+    color: white !important;
+    border: none !important;
+    transition: all 0.3s ease !important;
+}
+.button:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 4px 8px rgba(255, 105, 180, 0.3);
+}
+"""
+
+# Create the Gradio interface
+with gr.Blocks(theme="soft", css=custom_css) as demo:
+    gr.Markdown(f"# {title}")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        with gr.Column():
+            vibe_dropdown = gr.Dropdown(
+                choices=[
+                    "romantic",
+                    "cheesy",
+                    "nerdy",
+                    "cringe",
+                    "flirty"
+                ],
+                label="Pick a vibe",
+                value="romantic"
+            )
+            generate_btn = gr.Button("Generate Line", elem_classes="button")
+        
+        with gr.Column():
+            output = gr.Textbox(
+                label="Your pickup line",
+                lines=3,
+                interactive=False
+            )
+            copy_btn = gr.Button("📋 Copy to Clipboard", elem_classes="button")
+    
+    # Example inputs
+    gr.Examples(
+        examples=[
+            ["romantic"],
+            ["cheesy"],
+            ["nerdy"],
+            ["cringe"],
+            ["flirty"]
+        ],
+        inputs=[vibe_dropdown]
+    )
+    
+    generate_btn.click(
+        fn=generate_pickup_line,
+        inputs=[vibe_dropdown],
+        outputs=output
+    )
+    
+    # Footer
+    gr.Markdown("""
+    <div style="text-align: center; margin-top: 20px; color: #666;">
+        Built by Nath with SmolLM 🔥
+    </div>
+    """)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch(share=True)  # Set share=False if you don't want to create a public link