Upload folder using huggingface_hub

.eslintrc.json

@@ -0,0 +1,3 @@
+{
+  "extends": ["next/core-web-vitals", "next/typescript"]
+}

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+public/og-image.png filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,37 @@
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+.yarn/install-state.gz
+
+# testing
+/coverage
+
+# next.js
+/.next/
+/out/
+.env
+
+# production
+/build
+
+# misc
+.DS_Store
+*.pem
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+
+# local env files
+.env*.local
+
+# vercel
+.vercel
+
+# typescript
+*.tsbuildinfo
+next-env.d.ts

.prettierrc

@@ -0,0 +1 @@
+{"plugins": ["prettier-plugin-tailwindcss"]}

app/api/generate-logo/route.ts

@@ -0,0 +1,146 @@
+import { z } from "zod";
+import { Ratelimit } from "@upstash/ratelimit";
+import { Redis } from "@upstash/redis";
+import dedent from 'dedent';
+
+let ratelimit: Ratelimit | undefined;
+
+export async function POST(req: Request) {
+  const json = await req.json();
+    const data = z
+      .object({
+        userAPIKey: z.string().optional(),
+        companyName: z.string(),
+        selectedStyle: z.string(),
+        selectedPrimaryColor: z.string(),
+        selectedBackgroundColor: z.string(),
+        additionalInfo: z.string().optional(),
+      })
+      .parse(json);
+
+    // Add rate limiting if Upstash API keys are set & no BYOK, otherwise skip
+    if (process.env.UPSTASH_REDIS_REST_URL && !data.userAPIKey) {
+      ratelimit = new Ratelimit({
+        redis: Redis.fromEnv(),
+        // Allow 3 requests per 2 months on prod
+        limiter: Ratelimit.fixedWindow(3, "60 d"),
+        analytics: true,
+        prefix: "logocreator",
+      });
+    }
+
+  const apiKey = data.userAPIKey || process.env.SILICONFLOW_API_KEY;
+  const headers = {
+    "Authorization": `Bearer ${apiKey}`,
+    "Content-Type": "application/json",
+  };
+
+  if (!apiKey) {
+    return new Response("Missing Siliconflow API Key", {
+      status: 401,
+      headers: { "Content-Type": "text/plain" },
+    });
+  }
+
+
+  if (ratelimit) {
+    const identifier = "anonymous";
+    const { success, remaining } = await ratelimit.limit(identifier);
+
+    if (!success) {
+      return new Response(
+        "You've used up all your free credits. Please wait or try again later.",
+        {
+          status: 429,
+          headers: { "Content-Type": "text/plain" },
+        },
+      );
+    }
+  }
+
+      const flashyStyle =
+        "Flashy, attention grabbing, bold, futuristic, and eye-catching. Use vibrant neon colors with metallic, shiny, and glossy accents.";
+
+    const techStyle =
+        "highly detailed, sharp focus, cinematic, photorealistic, Minimalist, clean, sleek, neutral color pallete with subtle accents, clean lines, shadows, and flat.";
+
+    const modernStyle =
+        "modern, forward-thinking, flat design, geometric shapes, clean lines, natural colors with subtle accents, use strategic negative space to create visual interest.";
+
+    const playfulStyle =
+        "playful, lighthearted, bright bold colors, rounded shapes, lively.";
+
+    const abstractStyle =
+        "abstract, artistic, creative, unique shapes, patterns, and textures to create a visually interesting and wild logo.";
+
+    const minimalStyle =
+        "minimal, simple, timeless, versatile, single color logo, use negative space, flat design with minimal details, Light, soft, and subtle.";
+
+      const styleLookup: Record<string, string> = {
+        Flashy: flashyStyle,
+        Tech: techStyle,
+        Modern: modernStyle,
+        Playful: playfulStyle,
+        Abstract: abstractStyle,
+        Minimal: minimalStyle,
+      };
+
+    const prompt = dedent`A single logo, high-quality, award-winning professional design, made for both digital and print media, only contains a few vector shapes, ${styleLookup[data.selectedStyle]}
+
+      Primary color is ${data.selectedPrimaryColor.toLowerCase()} and background color is ${data.selectedBackgroundColor.toLowerCase()}. The company name is ${data.companyName}, make sure to include the company name in the logo. ${data.additionalInfo ? `Additional info: ${data.additionalInfo}` : ""}`;
+
+      interface SiliconFlowPayload {
+        model: string;
+        prompt: string;
+        image_size: string;
+        batch_size: number;
+        num_inference_steps: number;
+        guidance_scale: number;
+      }
+
+        const payload: SiliconFlowPayload = {
+            model: "black-forest-labs/FLUX.1-schnell",
+            prompt,
+            image_size: "1024x1024",
+            batch_size: 1,
+            num_inference_steps: 20,
+            guidance_scale: 7.5,
+          };
+      
+
+    try {
+      const response = await fetch(
+        "https://api.siliconflow.cn/v1/images/generations",
+        {
+          method: "POST",
+          headers,
+          body: JSON.stringify(payload),
+        }
+      );
+
+       if (!response.ok) {
+             const errorText = await response.text()
+            const message = `Siliconflow API Error - Status: ${response.status}, Text: ${errorText}`
+              return new Response(message, {
+                status: response.status,
+                headers: { "Content-Type": "text/plain" },
+              });
+            }
+
+
+       const json = await response.json() as any;
+       return Response.json(
+            { url: json.images[0].url, }, { status: 200 }
+        );
+     } catch (error) {
+        return new Response(
+            `Failed to call SiliconFlow API, ${error}`,
+            {
+              status: 500,
+              headers: { "Content-Type": "text/plain" },
+            },
+        );
+     }
+  }
+
+export const runtime = "edge";

app/components/Footer.tsx

@@ -0,0 +1,20 @@
+import Image from "next/image";
+
+const Footer = () => (
+  <div className="flex flex-col items-center justify-between bg-[#343434] px-4 py-4 md:flex-row">
+    <div className="mb-2 flex-grow pl-0 text-center md:mb-0 md:pl-0 lg:pl-48">
+      <span className="text-sm text-[#6F6F6F]">
+        Powered by{" "}
+        <a href="https://dub.sh/together-ai" className="underline">
+          Together.ai
+        </a>{" "}
+        &{" "}
+        <a href="https://dub.sh/flux-playground" className="underline">
+          Flux
+        </a>
+      </span>
+    </div>
+  </div>
+);
+
+export default Footer;

app/components/Header.tsx

@@ -0,0 +1,24 @@
+import Image from "next/image";
+import Link from "next/link";
+
+
+export default function Header({ className }: { className: string }) {
+  return (
+    <header className={`relative w-full ${className}`}>
+      <div className="flex items-center justify-between bg-[#343434] px-4 py-2 md:mt-4">
+          <div className="flex flex-grow justify-start xl:justify-center">
+            <Link href="https://dub.sh/together-ai" className="flex items-center">
+            <Image
+              src="/together-ai-logo1.svg"
+              alt="together.ai"
+              width={400}
+              height={120}
+              className="w-[220px] md:w-[330px] lg:w-[390px]"
+              priority
+              />
+          </Link>
+        </div>
+      </div>
+    </header>
+  );
+}

app/components/InfoToolTip.tsx

@@ -0,0 +1,9 @@
+import { Info } from "lucide-react";
+
+export default function InfoTooltip({ content }: { content: string }) {
+  return (
+    <span className="ml-2 cursor-default text-[#6F6F6F]" title={content}>
+        <Info size={11} />
+    </span>
+  );
+}

app/components/Spinner.tsx

@@ -0,0 +1,54 @@
+import { ReactNode } from "react";
+
+export default function Spinner({
+  loading = true,
+  children,
+  className = "",
+}: {
+  loading?: boolean;
+  children?: ReactNode;
+  className?: string;
+}) {
+  if (!loading) return children;
+
+  const spinner = (
+    <>
+      <style jsx>
+        {`
+          @keyframes spinner {
+            from {
+              opacity: 1;
+            }
+            to {
+              opacity: 0.25;
+            }
+          }
+        `}
+      </style>
+      <span className={`relative inline-flex size-4 ${className}`}>
+        {Array.from(Array(8).keys()).map((i) => (
+          <span
+            key={i}
+            className="absolute left-[calc(50%-12.5%/2)] top-0 h-[100%] w-[12.5%] animate-[spinner_800ms_linear_infinite] before:block before:h-[30%] before:w-[100%] before:rounded-full before:bg-current"
+            style={{
+              transform: `rotate(${45 * i}deg)`,
+              animationDelay: `calc(-${8 - i} / 8 * 800ms)`,
+            }}
+          />
+        ))}
+      </span>
+    </>
+  );
+
+  if (!children) return spinner;
+
+  return (
+    <span className="relative flex h-full items-center justify-center">
+      <span className="invisible">{children}</span>
+
+      <span className="absolute inset-0 flex items-center justify-center">
+        {spinner}
+      </span>
+    </span>
+  );
+}

app/components/ui/button.tsx

@@ -0,0 +1,57 @@
+import * as React from "react";
+import { Slot } from "@radix-ui/react-slot";
+import { cva, type VariantProps } from "class-variance-authority";
+
+import { cn } from "@/lib/utils";
+
+const buttonVariants = cva(
+  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-colors focus-visible:outline focus-visible:outline-offset-2 focus-visible:outline-white disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg]:size-4 [&_svg]:shrink-0",
+  {
+    variants: {
+      variant: {
+        default:
+          "bg-primary text-primary-foreground shadow hover:bg-primary/90",
+        destructive:
+          "bg-destructive text-destructive-foreground shadow-sm hover:bg-destructive/90",
+        outline:
+          "border border-input bg-background shadow-sm hover:bg-accent hover:text-accent-foreground",
+        secondary:
+          "bg-secondary text-secondary-foreground shadow-sm hover:bg-secondary/80",
+        ghost: "hover:bg-accent hover:text-accent-foreground",
+        link: "text-primary underline-offset-4 hover:underline",
+      },
+      size: {
+        default: "h-9 px-4 py-2",
+        sm: "h-8 rounded-md px-3 text-xs",
+        lg: "h-12 rounded-md px-8",
+        icon: "h-9 w-9",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+      size: "default",
+    },
+  },
+);
+
+export interface ButtonProps
+  extends React.ButtonHTMLAttributes<HTMLButtonElement>,
+    VariantProps<typeof buttonVariants> {
+  asChild?: boolean;
+}
+
+const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  ({ className, variant, size, asChild = false, ...props }, ref) => {
+    const Comp = asChild ? Slot : "button";
+    return (
+      <Comp
+        className={cn(buttonVariants({ variant, size, className }))}
+        ref={ref}
+        {...props}
+      />
+    );
+  },
+);
+Button.displayName = "Button";
+
+export { Button, buttonVariants };

app/components/ui/input.tsx

@@ -0,0 +1,22 @@
+import * as React from "react";
+
+import { cn } from "@/lib/utils";
+
+const Input = React.forwardRef<HTMLInputElement, React.ComponentProps<"input">>(
+  ({ className, type, ...props }, ref) => {
+    return (
+      <input
+        type={type}
+        className={cn(
+          "flex h-11 w-full rounded-md border border-input bg-[#343434] px-3 py-1 text-base shadow-sm transition-colors file:border-0 file:bg-transparent file:text-sm file:font-medium file:text-foreground placeholder:text-muted-foreground focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:cursor-not-allowed disabled:opacity-50 md:text-sm",
+          className,
+        )}
+        ref={ref}
+        {...props}
+      />
+    );
+  },
+);
+Input.displayName = "Input";
+
+export { Input };

app/components/ui/select.tsx

@@ -0,0 +1,159 @@
+"use client";
+
+import * as React from "react";
+import * as SelectPrimitive from "@radix-ui/react-select";
+import { Check, ChevronDown, ChevronUp } from "lucide-react";
+
+import { cn } from "@/lib/utils";
+
+const Select = SelectPrimitive.Root;
+
+const SelectGroup = SelectPrimitive.Group;
+
+const SelectValue = SelectPrimitive.Value;
+
+const SelectTrigger = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.Trigger>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.Trigger>
+>(({ className, children, ...props }, ref) => (
+  <SelectPrimitive.Trigger
+    ref={ref}
+    className={cn(
+      "flex h-9 w-full items-center justify-between whitespace-nowrap rounded-md border border-input bg-[#343434] px-3 py-2 text-sm ring-offset-background placeholder:text-muted-foreground focus:outline-none focus:ring-1 focus:ring-ring disabled:cursor-not-allowed disabled:opacity-50 [&>span]:line-clamp-1",
+      className,
+    )}
+    {...props}
+  >
+    {children}
+    <SelectPrimitive.Icon asChild>
+      <ChevronDown className="h-4 w-4 opacity-50" />
+    </SelectPrimitive.Icon>
+  </SelectPrimitive.Trigger>
+));
+SelectTrigger.displayName = SelectPrimitive.Trigger.displayName;
+
+const SelectScrollUpButton = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.ScrollUpButton>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.ScrollUpButton>
+>(({ className, ...props }, ref) => (
+  <SelectPrimitive.ScrollUpButton
+    ref={ref}
+    className={cn(
+      "flex cursor-default items-center justify-center py-1",
+      className,
+    )}
+    {...props}
+  >
+    <ChevronUp className="h-4 w-4" />
+  </SelectPrimitive.ScrollUpButton>
+));
+SelectScrollUpButton.displayName = SelectPrimitive.ScrollUpButton.displayName;
+
+const SelectScrollDownButton = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.ScrollDownButton>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.ScrollDownButton>
+>(({ className, ...props }, ref) => (
+  <SelectPrimitive.ScrollDownButton
+    ref={ref}
+    className={cn(
+      "flex cursor-default items-center justify-center py-1",
+      className,
+    )}
+    {...props}
+  >
+    <ChevronDown className="h-4 w-4" />
+  </SelectPrimitive.ScrollDownButton>
+));
+SelectScrollDownButton.displayName =
+  SelectPrimitive.ScrollDownButton.displayName;
+
+const SelectContent = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.Content>
+>(({ className, children, position = "popper", ...props }, ref) => (
+  <SelectPrimitive.Portal>
+    <SelectPrimitive.Content
+      ref={ref}
+      className={cn(
+        "relative z-50 max-h-96 min-w-[8rem] overflow-hidden rounded-md border bg-popover text-popover-foreground shadow-md data-[state=open]:animate-in data-[state=closed]:animate-out data-[state=closed]:fade-out-0 data-[state=open]:fade-in-0 data-[state=closed]:zoom-out-95 data-[state=open]:zoom-in-95 data-[side=bottom]:slide-in-from-top-2 data-[side=left]:slide-in-from-right-2 data-[side=right]:slide-in-from-left-2 data-[side=top]:slide-in-from-bottom-2",
+        position === "popper" &&
+          "data-[side=bottom]:translate-y-1 data-[side=left]:-translate-x-1 data-[side=right]:translate-x-1 data-[side=top]:-translate-y-1",
+        className,
+      )}
+      position={position}
+      {...props}
+    >
+      <SelectScrollUpButton />
+      <SelectPrimitive.Viewport
+        className={cn(
+          "p-1",
+          position === "popper" &&
+            "h-[var(--radix-select-trigger-height)] w-full min-w-[var(--radix-select-trigger-width)]",
+        )}
+      >
+        {children}
+      </SelectPrimitive.Viewport>
+      <SelectScrollDownButton />
+    </SelectPrimitive.Content>
+  </SelectPrimitive.Portal>
+));
+SelectContent.displayName = SelectPrimitive.Content.displayName;
+
+const SelectLabel = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.Label>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.Label>
+>(({ className, ...props }, ref) => (
+  <SelectPrimitive.Label
+    ref={ref}
+    className={cn("px-2 py-1.5 text-sm font-semibold", className)}
+    {...props}
+  />
+));
+SelectLabel.displayName = SelectPrimitive.Label.displayName;
+
+const SelectItem = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.Item>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.Item>
+>(({ className, children, ...props }, ref) => (
+  <SelectPrimitive.Item
+    ref={ref}
+    className={cn(
+      "relative flex w-full cursor-default select-none items-center rounded-sm py-1.5 pl-2 pr-8 text-sm outline-none focus:bg-accent focus:text-accent-foreground data-[disabled]:pointer-events-none data-[disabled]:opacity-50",
+      className,
+    )}
+    {...props}
+  >
+    <span className="absolute right-2 flex h-3.5 w-3.5 items-center justify-center">
+      <SelectPrimitive.ItemIndicator>
+        <Check className="h-4 w-4" />
+      </SelectPrimitive.ItemIndicator>
+    </span>
+    <SelectPrimitive.ItemText>{children}</SelectPrimitive.ItemText>
+  </SelectPrimitive.Item>
+));
+SelectItem.displayName = SelectPrimitive.Item.displayName;
+
+const SelectSeparator = React.forwardRef<
+  React.ElementRef<typeof SelectPrimitive.Separator>,
+  React.ComponentPropsWithoutRef<typeof SelectPrimitive.Separator>
+>(({ className, ...props }, ref) => (
+  <SelectPrimitive.Separator
+    ref={ref}
+    className={cn("-mx-1 my-1 h-px bg-muted", className)}
+    {...props}
+  />
+));
+SelectSeparator.displayName = SelectPrimitive.Separator.displayName;
+
+export {
+  Select,
+  SelectGroup,
+  SelectValue,
+  SelectTrigger,
+  SelectContent,
+  SelectLabel,
+  SelectItem,
+  SelectSeparator,
+  SelectScrollUpButton,
+  SelectScrollDownButton,
+};

app/components/ui/textarea.tsx

@@ -0,0 +1,22 @@
+import * as React from "react";
+
+import { cn } from "@/lib/utils";
+
+const Textarea = React.forwardRef<
+  HTMLTextAreaElement,
+  React.ComponentProps<"textarea">
+>(({ className, ...props }, ref) => {
+  return (
+    <textarea
+      className={cn(
+        "flex min-h-[60px] w-full rounded-md border border-input bg-[#343434] px-3 py-2 text-base shadow-sm placeholder:text-muted-foreground focus-visible:outline-none focus-visible:ring-1 focus-visible:ring-ring disabled:cursor-not-allowed disabled:opacity-50 md:text-sm",
+        className,
+      )}
+      ref={ref}
+      {...props}
+    />
+  );
+});
+Textarea.displayName = "Textarea";
+
+export { Textarea };

app/components/ui/toast.tsx

@@ -0,0 +1,129 @@
+"use client";
+
+import * as React from "react";
+import * as ToastPrimitives from "@radix-ui/react-toast";
+import { cva, type VariantProps } from "class-variance-authority";
+import { X } from "lucide-react";
+
+import { cn } from "@/lib/utils";
+
+const ToastProvider = ToastPrimitives.Provider;
+
+const ToastViewport = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Viewport>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Viewport>
+>(({ className, ...props }, ref) => (
+  <ToastPrimitives.Viewport
+    ref={ref}
+    className={cn(
+      "fixed top-0 z-[100] flex max-h-screen w-full flex-col-reverse p-4 sm:right-0 sm:flex-col md:max-w-[420px]",
+      className,
+    )}
+    {...props}
+  />
+));
+ToastViewport.displayName = ToastPrimitives.Viewport.displayName;
+
+const toastVariants = cva(
+  "group pointer-events-auto relative flex w-full items-center justify-between space-x-2 overflow-hidden rounded-md border p-4 pr-6 shadow-lg transition-all data-[swipe=cancel]:translate-x-0 data-[swipe=end]:translate-x-[var(--radix-toast-swipe-end-x)] data-[swipe=move]:translate-x-[var(--radix-toast-swipe-move-x)] data-[swipe=move]:transition-none data-[state=open]:animate-in data-[state=closed]:animate-out data-[swipe=end]:animate-out data-[state=closed]:fade-out-80 data-[state=closed]:slide-out-to-right-full data-[state=open]:slide-in-from-top-full",
+  {
+    variants: {
+      variant: {
+        default: "border bg-background text-foreground",
+        destructive:
+          "destructive group border-destructive bg-destructive text-destructive-foreground",
+      },
+    },
+    defaultVariants: {
+      variant: "default",
+    },
+  },
+);
+
+const Toast = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Root>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Root> &
+    VariantProps<typeof toastVariants>
+>(({ className, variant, ...props }, ref) => {
+  return (
+    <ToastPrimitives.Root
+      ref={ref}
+      className={cn(toastVariants({ variant }), className)}
+      {...props}
+    />
+  );
+});
+Toast.displayName = ToastPrimitives.Root.displayName;
+
+const ToastAction = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Action>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Action>
+>(({ className, ...props }, ref) => (
+  <ToastPrimitives.Action
+    ref={ref}
+    className={cn(
+      "inline-flex h-8 shrink-0 items-center justify-center rounded-md border bg-transparent px-3 text-sm font-medium transition-colors hover:bg-secondary focus:outline-none focus:ring-1 focus:ring-ring disabled:pointer-events-none disabled:opacity-50 group-[.destructive]:border-muted/40 group-[.destructive]:hover:border-destructive/30 group-[.destructive]:hover:bg-destructive group-[.destructive]:hover:text-destructive-foreground group-[.destructive]:focus:ring-destructive",
+      className,
+    )}
+    {...props}
+  />
+));
+ToastAction.displayName = ToastPrimitives.Action.displayName;
+
+const ToastClose = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Close>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Close>
+>(({ className, ...props }, ref) => (
+  <ToastPrimitives.Close
+    ref={ref}
+    className={cn(
+      "absolute right-1 top-1 rounded-md p-1 text-foreground/50 opacity-0 transition-opacity hover:text-foreground focus:opacity-100 focus:outline-none focus:ring-1 group-hover:opacity-100 group-[.destructive]:text-red-300 group-[.destructive]:hover:text-red-50 group-[.destructive]:focus:ring-red-400 group-[.destructive]:focus:ring-offset-red-600",
+      className,
+    )}
+    toast-close=""
+    {...props}
+  >
+    <X className="h-4 w-4" />
+  </ToastPrimitives.Close>
+));
+ToastClose.displayName = ToastPrimitives.Close.displayName;
+
+const ToastTitle = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Title>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Title>
+>(({ className, ...props }, ref) => (
+  <ToastPrimitives.Title
+    ref={ref}
+    className={cn("text-sm font-semibold [&+div]:text-xs", className)}
+    {...props}
+  />
+));
+ToastTitle.displayName = ToastPrimitives.Title.displayName;
+
+const ToastDescription = React.forwardRef<
+  React.ElementRef<typeof ToastPrimitives.Description>,
+  React.ComponentPropsWithoutRef<typeof ToastPrimitives.Description>
+>(({ className, ...props }, ref) => (
+  <ToastPrimitives.Description
+    ref={ref}
+    className={cn("text-sm opacity-90", className)}
+    {...props}
+  />
+));
+ToastDescription.displayName = ToastPrimitives.Description.displayName;
+
+type ToastProps = React.ComponentPropsWithoutRef<typeof Toast>;
+
+type ToastActionElement = React.ReactElement<typeof ToastAction>;
+
+export {
+  type ToastProps,
+  type ToastActionElement,
+  ToastProvider,
+  ToastViewport,
+  Toast,
+  ToastTitle,
+  ToastDescription,
+  ToastClose,
+  ToastAction,
+};

app/globals.css

@@ -0,0 +1,68 @@
+@tailwind base;
+@tailwind components;
+@tailwind utilities;
+
+@layer base {
+  :root {
+    --background: 0 0% 100%;
+    --foreground: 0 0% 3.9%;
+    --card: 0 0% 100%;
+    --card-foreground: 0 0% 3.9%;
+    --popover: 0 0% 100%;
+    --popover-foreground: 0 0% 3.9%;
+    --primary: 0 0% 9%;
+    --primary-foreground: 0 0% 98%;
+    --secondary: 0 0% 96.1%;
+    --secondary-foreground: 0 0% 9%;
+    --muted: 0 0% 96.1%;
+    --muted-foreground: 0 0% 45.1%;
+    --accent: 0 0% 96.1%;
+    --accent-foreground: 0 0% 9%;
+    --destructive: 0 84.2% 60.2%;
+    --destructive-foreground: 0 0% 98%;
+    --border: 0 0% 89.8%;
+    --input: 0 0% 89.8%;
+    --ring: 0 0% 3.9%;
+    --chart-1: 12 76% 61%;
+    --chart-2: 173 58% 39%;
+    --chart-3: 197 37% 24%;
+    --chart-4: 43 74% 66%;
+    --chart-5: 27 87% 67%;
+    --radius: 0.5rem;
+  }
+  .dark {
+    --background: 0 0% 3.9%;
+    --foreground: 0 0% 98%;
+    --card: 0 0% 3.9%;
+    --card-foreground: 0 0% 98%;
+    --popover: 0 0% 3.9%;
+    --popover-foreground: 0 0% 98%;
+    --primary: 0 0% 98%;
+    --primary-foreground: 0 0% 9%;
+    --secondary: 0 0% 14.9%;
+    --secondary-foreground: 0 0% 98%;
+    --muted: 0 0% 14.9%;
+    --muted-foreground: 0 0% 63.9%;
+    --accent: 0 0% 14.9%;
+    --accent-foreground: 0 0% 98%;
+    --destructive: 0 62.8% 30.6%;
+    --destructive-foreground: 0 0% 98%;
+    --border: 0 0% 20.39%;
+    --input: 0 0% 20.39%;
+    --ring: 0 0% 83.1%;
+    --chart-1: 220 70% 50%;
+    --chart-2: 160 60% 45%;
+    --chart-3: 30 80% 55%;
+    --chart-4: 280 65% 60%;
+    --chart-5: 340 75% 55%;
+  }
+}
+
+@layer base {
+  * {
+    @apply border-border;
+  }
+  body {
+    @apply bg-background text-foreground;
+  }
+}

app/layout.tsx

@@ -0,0 +1,58 @@
+import type { Metadata } from "next";
+import { Jura } from "next/font/google";
+import "./globals.css";
+import { Toaster } from "@/app/components/ui/toaster";
+
+const jura = Jura({
+  subsets: ["latin"],
+  variable: "--font-jura",
+});
+
+const title = "Logo-creator.io – Generate a logo";
+const description = "Generate a logo for your company";
+const url = "https://www.logo-creator.io/";
+const ogimage = "https://www.logo-creator.io/og-image.png";
+const sitename = "logo-creator.io";
+
+export const metadata: Metadata = {
+  metadataBase: new URL(url),
+  title,
+  description,
+  icons: {
+    icon: "/favicon.ico",
+  },
+  openGraph: {
+    images: [ogimage],
+    title,
+    description,
+    url: url,
+    siteName: sitename,
+    locale: "en_US",
+    type: "website",
+  },
+  twitter: {
+    card: "summary_large_image",
+    images: [ogimage],
+    title,
+    description,
+  },
+};
+
+export default function RootLayout({
+  children,
+}: Readonly<{
+  children: React.ReactNode;
+}>) {
+  return (
+    <html lang="en" className="h-full">
+      <head>
+         <meta name="color-scheme" content="dark" />
+      </head>
+      <body
+        className={`${jura.variable} dark min-h-full bg-[#343434] font-jura antialiased`}
+      >
+        {children}
+      </body>
+    </html>
+  );
+}

app/lib/domain.ts

@@ -0,0 +1,8 @@
+export const domain =
+  process.env.NEXT_PUBLIC_VERCEL_ENV === "production"
+    ? "https://www.logo-creator.io"
+    : process.env.VERCEL_BRANCH_URL
+      ? `https://${process.env.VERCEL_BRANCH_URL}`
+      : process.env.NEXT_PUBLIC_VERCEL_URL
+        ? `https://${process.env.NEXT_PUBLIC_VERCEL_URL}`
+        : "http://localhost:3000";

app/lib/utils.ts

@@ -0,0 +1,6 @@
+import { type ClassValue, clsx } from "clsx";
+import { twMerge } from "tailwind-merge";
+
+export function cn(...inputs: ClassValue[]) {
+  return twMerge(clsx(inputs));
+}

app/page.tsx

@@ -0,0 +1,296 @@
+"use client";
+
+import Spinner from "@/app/components/Spinner";
+import { Button } from "@/app/components/ui/button";
+import { Input } from "@/app/components/ui/input";
+import { Textarea } from "@/app/components/ui/textarea";
+import { motion } from "framer-motion";
+import {
+  Select,
+  SelectContent,
+  SelectGroup,
+  SelectItem,
+  SelectTrigger,
+  SelectValue,
+} from "@/components/ui/select";
+import * as RadioGroup from "@radix-ui/react-radio-group";
+import { DownloadIcon, RefreshCwIcon } from "lucide-react";
+import Image from "next/image";
+import { useState } from "react";
+import Header from "./components/Header";
+import Footer from "./components/Footer";
+import { domain } from "@/app/lib/domain";
+
+const logoStyles = [
+  { name: "科技", icon: "/tech.svg" },
+  { name: "炫彩", icon: "/flashy.svg" },
+  { name: "现代", icon: "/modern.svg" },
+  { name: "趣味", icon: "/playful.svg" },
+  { name: "抽象", icon: "/abstract.svg" },
+  { name: "简约", icon: "/minimal.svg" },
+];
+
+const primaryColors = [
+  { name: "蓝色", color: "#0F6FFF" },
+  { name: "红色", color: "#FF0000" },
+  { name: "绿色", color: "#00FF00" },
+  { name: "黄色", color: "#FFFF00" },
+];
+
+const backgroundColors = [
+  { name: "白色", color: "#FFFFFF" },
+  { name: "灰色", color: "#CCCCCC" },
+  { name: "黑色", color: "#000000" },
+];
+
+export default function Page() {
+  const [companyName, setCompanyName] = useState("");
+  const [selectedStyle, setSelectedStyle] = useState(logoStyles[0].name);
+  const [selectedPrimaryColor, setSelectedPrimaryColor] = useState(
+    primaryColors[0].name,
+  );
+  const [selectedBackgroundColor, setSelectedBackgroundColor] = useState(
+    backgroundColors[0].name,
+  );
+  const [additionalInfo, setAdditionalInfo] = useState("");
+  const [isLoading, setIsLoading] = useState(false);
+  const [generatedImage, setGeneratedImage] = useState("");
+
+  async function generateLogo() {
+    setIsLoading(true);
+
+    const res = await fetch("/api/generate-logo", {
+      method: "POST",
+      body: JSON.stringify({
+        companyName,
+        selectedStyle,
+        selectedPrimaryColor,
+        selectedBackgroundColor,
+        additionalInfo,
+      }),
+    });
+
+    if (res.ok) {
+      const json = await res.json() as any;
+      setGeneratedImage(json.url);
+    } else {
+      alert(`Error: ${res.statusText} `);
+    }
+    setIsLoading(false);
+  }
+
+  return (
+    <div className="flex h-screen flex-col overflow-y-auto overflow-x-hidden bg-[#343434] md:flex-row">
+      <Header className="block md:hidden" />
+
+      <div className="flex w-full flex-col md:flex-row">
+        <div className="relative flex h-full w-full flex-col bg-[#2C2C2C] text-[#F3F3F3] md:max-w-sm">
+          <form
+            onSubmit={(e) => {
+              e.preventDefault();
+              setGeneratedImage("");
+              generateLogo();
+            }}
+            className="flex h-full w-full flex-col"
+          >
+            <fieldset className="flex grow flex-col">
+              <div className="flex-grow overflow-y-auto">
+                <div className="px-8 pb-0 pt-4 md:px-6 md:pt-6">
+                  <div className="mb-6">
+                    <label
+                      htmlFor="company-name"
+                      className="mb-2 block text-xs font-bold uppercase text-[#6F6F6F]"
+                    >
+                     公司名称
+                    </label>
+                    <Input
+                      value={companyName}
+                      onChange={(e) => setCompanyName(e.target.value)}
+                      placeholder="例如：Sam's Burgers"
+                      required
+                    />
+                  </div>
+
+                  <div className="mb-6">
+                    <label className="mb-2 flex items-center text-xs font-bold uppercase text-[#6F6F6F]">
+                      风格
+                    </label>
+                    <RadioGroup.Root
+                      value={selectedStyle}
+                      onValueChange={setSelectedStyle}
+                      className="grid grid-cols-3 gap-3"
+                    >
+                      {logoStyles.map((logoStyle) => (
+                        <RadioGroup.Item
+                          value={logoStyle.name}
+                          key={logoStyle.name}
+                          className="group text-[#6F6F6F] focus-visible:outline-none data-[state=checked]:text-white"
+                        >
+                          <Image
+                            src={logoStyle.icon}
+                            alt={logoStyle.name}
+                            width={96}
+                            height={96}
+                            className="w-full rounded-md border border-transparent group-focus-visible:outline group-focus-visible:outline-offset-2 group-focus-visible:outline-gray-400 group-data-[state=checked]:border-white"
+                          />
+                          <span className="text-xs">{logoStyle.name}</span>
+                        </RadioGroup.Item>
+                      ))}
+                    </RadioGroup.Root>
+                  </div>
+
+                  <div className="mb-[25px] flex flex-col md:flex-row md:space-x-3">
+                    <div className="mb-4 flex-1 md:mb-0">
+                      <label className="mb-1 block text-xs font-bold uppercase text-[#6F6F6F]">
+                       主要颜色
+                      </label>
+                      <Select
+                        value={selectedPrimaryColor}
+                        onValueChange={setSelectedPrimaryColor}
+                      >
+                        <SelectTrigger>
+                          <SelectValue placeholder="选择颜色" />
+                        </SelectTrigger>
+                        <SelectContent>
+                          <SelectGroup>
+                            {primaryColors.map((color) => (
+                              <SelectItem key={color.color} value={color.name}>
+                                <span className="flex items-center">
+                                  <span
+                                    style={{ backgroundColor: color.color }}
+                                    className="mr-2 size-4 rounded-sm bg-white"
+                                  />
+                                  {color.name}
+                                </span>
+                              </SelectItem>
+                            ))}
+                          </SelectGroup>
+                        </SelectContent>
+                      </Select>
+                    </div>
+                    <div className="flex-1">
+                      <label className="mb-1 block items-center text-xs font-bold uppercase text-[#6F6F6F]">
+                        背景色
+                      </label>
+                      <Select
+                        value={selectedBackgroundColor}
+                        onValueChange={setSelectedBackgroundColor}
+                      >
+                        <SelectTrigger>
+                          <SelectValue placeholder="选择颜色" />
+                        </SelectTrigger>
+                        <SelectContent>
+                          <SelectGroup>
+                            {backgroundColors.map((color) => (
+                              <SelectItem key={color.color} value={color.name}>
+                                <span className="flex items-center">
+                                  <span
+                                    style={{ backgroundColor: color.color }}
+                                    className="mr-2 size-4 rounded-sm bg-white"
+                                  />
+                                  {color.name}
+                                </span>
+                              </SelectItem>
+                            ))}
+                          </SelectGroup>
+                        </SelectContent>
+                      </Select>
+                    </div>
+                  </div>
+                  <div className="mb-1">
+                    <div className="mt-1">
+                      <div className="mb-1">
+                        <label
+                          htmlFor="additional-info"
+                          className="mb-2 flex items-center text-xs font-bold uppercase text-[#6F6F6F]"
+                        >
+                         附加信息
+                        </label>
+                        <Textarea
+                          value={additionalInfo}
+                          onChange={(e) => setAdditionalInfo(e.target.value)}
+                          placeholder="输入关于你的logo的其他信息"
+                        />
+                      </div>
+                    </div>
+                  </div>
+                </div>
+              </div>
+              <div className="px-8 py-4 md:px-6 md:py-6">
+                <Button
+                  size="lg"
+                  className="w-full text-base font-bold"
+                  type="submit"
+                  disabled={isLoading}
+                >
+                  {isLoading ? (
+                    <div className="loader mr-2" />
+                  ) : (
+                    <Image
+                      src="/generate-icon.svg"
+                      alt="Generate Icon"
+                      width={16}
+                      height={16}
+                      className="mr-2"
+                    />
+                  )}
+                  {isLoading ? "生成中..." : "生成Logo"}
+                </Button>
+              </div>
+            </fieldset>
+          </form>
+        </div>
+
+        <div className="flex w-full flex-col pt-12 md:pt-0">
+          <Header className="hidden md:block" />
+           <div className="relative flex flex-grow items-center justify-center px-4">
+            <div className="relative aspect-square w-full max-w-lg">
+              {generatedImage ? (
+                <>
+                  <Image
+                    className={`${isLoading ? "animate-pulse" : ""}`}
+                    width={512}
+                    height={512}
+                    src={generatedImage}
+                    alt="Generated logo"
+                  />
+                  <div
+                    className={`pointer-events-none absolute inset-0 transition ${isLoading ? "bg-black/50 duration-500" : "bg-black/0 duration-0"}`}
+                  />
+
+                  <div className="absolute -right-12 top-0 flex flex-col gap-2">
+                    <Button size="icon" variant="secondary" asChild>
+                      <a href={generatedImage} download="logo.png">
+                        <DownloadIcon />
+                      </a>
+                    </Button>
+                    <Button
+                      size="icon"
+                      onClick={generateLogo}
+                      variant="secondary"
+                    >
+                      <Spinner loading={isLoading}>
+                        <RefreshCwIcon />
+                      </Spinner>
+                    </Button>
+                  </div>
+                </>
+              ) : (
+                <Spinner loading={isLoading} className="size-8 text-white">
+                  <div className="flex aspect-square w-full flex-col items-center justify-center rounded-xl bg-[#2C2C2C]">
+                    <h4 className="text-center text-base leading-tight text-white">
+                      在10秒内
+                      <br />
+                     生成你的专属 Logo
+                    </h4>
+                  </div>
+                </Spinner>
+              )}
+            </div>
+          </div>
+          <Footer />
+        </div>
+      </div>
+    </div>
+  );
+}

components.json

@@ -0,0 +1,21 @@
+{
+  "$schema": "https://ui.shadcn.com/schema.json",
+  "style": "new-york",
+  "rsc": true,
+  "tsx": true,
+  "tailwind": {
+    "config": "tailwind.config.ts",
+    "css": "app/globals.css",
+    "baseColor": "neutral",
+    "cssVariables": true,
+    "prefix": ""
+  },
+  "aliases": {
+    "components": "@/components",
+    "utils": "@/lib/utils",
+    "ui": "@/components/ui",
+    "lib": "@/lib",
+    "hooks": "@/hooks"
+  },
+  "iconLibrary": "lucide"
+}

lib/utils.ts

@@ -0,0 +1,6 @@
+import { clsx, type ClassValue } from "clsx";
+import { twMerge } from "tailwind-merge";
+
+export function cn(...inputs: ClassValue[]) {
+  return twMerge(clsx(inputs));
+}

middleware.ts

@@ -0,0 +1,23 @@
+import { NextResponse, NextFetchEvent } from "next/server";
+import type { NextRequest } from "next/server";
+
+export default async function middleware(
+  req: NextRequest,
+  evt: NextFetchEvent,
+) {
+  const country = req.geo?.country;
+  // Check for Russian traffic first as there's too much spam from Russia
+  if (country === "RU") {
+    return new NextResponse("Access Denied", { status: 403 });
+  }
+
+  // If not from Russia, proceed without any further authentication
+    return NextResponse.next();
+}
+
+export const config = {
+  matcher: [
+    // Skip Next.js internals and static files
+    "/((?!_next|[^?]*\\.(?:html?|css|js(?!on)|jpe?g|webp|png|gif|svg|ttf|woff2?|ico|csv|docx?|xlsx?|zip|webmanifest)).*)",
+  ],
+};

next.config.mjs

@@ -0,0 +1,13 @@
+/** @type {import('next').NextConfig} */
+const nextConfig = {
+  images: {
+    remotePatterns: [
+        {
+          protocol: 'https',
+          hostname: 'sc-maas.oss-cn-shanghai.aliyuncs.com',
+        },
+     ],
+  },
+};
+
+export default nextConfig;

package.json

@@ -0,0 +1,47 @@
+{
+  "name": "logo-maker",
+  "version": "0.1.0",
+  "private": true,
+  "scripts": {
+    "dev": "next dev",
+    "build": "next build",
+    "start": "next start",
+    "lint": "next lint"
+  },
+  "dependencies": {
+    "@radix-ui/react-dialog": "^1.1.2",
+    "@radix-ui/react-radio-group": "^1.2.1",
+    "@radix-ui/react-select": "^2.1.2",
+    "@radix-ui/react-slot": "^1.1.0",
+    "@radix-ui/react-toast": "^1.2.2",
+    "@radix-ui/react-tooltip": "^1.1.3",
+    "@upstash/ratelimit": "^2.0.4",
+    "axios": "^1.7.7",
+    "class-variance-authority": "^0.7.0",
+    "clsx": "^2.1.1",
+    "dedent": "^1.5.3",
+    "form-data-encoder": "1.7.2",
+    "framer-motion": "^11.11.15",
+    "lucide-react": "^0.453.0",
+    "next": "^14.2.15",
+    "next-plausible": "^3.12.4",
+    "react": "^18",
+    "react-dom": "^18",
+    "shadcn": "^1.0.0",
+    "tailwind-merge": "^2.5.4",
+    "tailwindcss-animate": "^1.0.7"
+  },
+  "devDependencies": {
+    "@shadcn/ui": "^0.0.4",
+    "@types/node": "^20",
+    "@types/react": "^18",
+    "@types/react-dom": "^18",
+    "eslint": "^8",
+    "eslint-config-next": "14.2.15",
+    "postcss": "^8",
+    "prettier": "^3.3.3",
+    "prettier-plugin-tailwindcss": "^0.6.8",
+    "tailwindcss": "^3.4.1",
+    "typescript": "^5"
+  }
+}

postcss.config.mjs

@@ -0,0 +1,8 @@
+/** @type {import('postcss-load-config').Config} */
+const config = {
+  plugins: {
+    tailwindcss: {},
+  },
+};
+
+export default config;

tailwind.config.ts

@@ -0,0 +1,75 @@
+import type { Config } from "tailwindcss";
+import tailwindAnimate from "tailwindcss-animate";
+
+const config: Config = {
+  darkMode: ["class"],
+  content: [
+    "./pages/**/*.{js,ts,jsx,tsx,mdx}",
+    "./components/**/*.{js,ts,jsx,tsx,mdx}",
+    "./app/**/*.{js,ts,jsx,tsx,mdx}",
+  ],
+  theme: {
+    extend: {
+      colors: {
+        gray: {
+          "700": "#2C2C2C",
+          "800": "#343434",
+          "900": "#1F1F1F",
+        },
+        blue: {
+          "500": "#0F6FFF",
+        },
+        background: "hsl(var(--background))",
+        foreground: "hsl(var(--foreground))",
+        card: {
+          DEFAULT: "hsl(var(--card))",
+          foreground: "hsl(var(--card-foreground))",
+        },
+        popover: {
+          DEFAULT: "hsl(var(--popover))",
+          foreground: "hsl(var(--popover-foreground))",
+        },
+        primary: {
+          DEFAULT: "hsl(var(--primary))",
+          foreground: "hsl(var(--primary-foreground))",
+        },
+        secondary: {
+          DEFAULT: "hsl(var(--secondary))",
+          foreground: "hsl(var(--secondary-foreground))",
+        },
+        muted: {
+          DEFAULT: "hsl(var(--muted))",
+          foreground: "hsl(var(--muted-foreground))",
+        },
+        accent: {
+          DEFAULT: "hsl(var(--accent))",
+          foreground: "hsl(var(--accent-foreground))",
+        },
+        destructive: {
+          DEFAULT: "hsl(var(--destructive))",
+          foreground: "hsl(var(--destructive-foreground))",
+        },
+        border: "hsl(var(--border))",
+        input: "hsl(var(--input))",
+        ring: "hsl(var(--ring))",
+        chart: {
+          "1": "hsl(var(--chart-1))",
+          "2": "hsl(var(--chart-2))",
+          "3": "hsl(var(--chart-3))",
+          "4": "hsl(var(--chart-4))",
+          "5": "hsl(var(--chart-5))",
+        },
+      },
+      fontFamily: {
+        jura: ["var(--font-jura)"],
+      },
+      borderRadius: {
+        lg: "var(--radius)",
+        md: "calc(var(--radius) - 2px)",
+        sm: "calc(var(--radius) - 4px)",
+      },
+    },
+  },
+  plugins: [tailwindAnimate],
+};
+export default config;

tsconfig.json

@@ -0,0 +1,27 @@
+{
+  "compilerOptions": {
+    "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": {
+      "@/*": ["./*"],
+      "@/components/*": ["./app/components/*"]
+    }
+  },
+  "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", ".next/types/**/*.ts"],
+  "exclude": ["node_modules"]
+}

整理项目结构.py

@@ -0,0 +1,64 @@
+import os
+
+def collect_code_files_to_txt(
+    repo_path: str,
+    output_file: str,
+    exclude_extensions=None,
+    encoding='utf-8'
+):
+    """
+    将指定目录及其子目录下的所有文件内容整合到同一个文本文件中，
+    但会排除某些不想处理的后缀（比如图片、音频等）。
+    
+    :param repo_path:    项目根目录路径
+    :param output_file:  输出的txt文件路径
+    :param exclude_extensions:   需要排除的文件后缀列表(如 ['.svg','.png','.jpg']),
+    :param encoding:     打开文件使用的编码，默认为 utf-8。
+    """
+
+    with open(output_file, 'w', encoding=encoding) as out_f:
+        for root, dirs, files in os.walk(repo_path):
+            # 排除一些目录
+            if "__pycache__" in dirs:
+                dirs.remove("__pycache__")
+            if "node_modules" in dirs:
+                dirs.remove("node_modules")
+            if "models" in dirs:
+                dirs.remove("models")
+
+            for filename in files:
+                # 如果排除列表不为空，则检查是否需要跳过
+                if exclude_extensions is not None:
+                    _, ext = os.path.splitext(filename)
+                    if ext.lower() in exclude_extensions:
+                        continue  # 跳过这些扩展名
+
+                full_path = os.path.join(root, filename)
+
+                try:
+                    with open(full_path, 'r', encoding=encoding) as code_f:
+                        content = code_f.read()
+                except Exception as e:
+                    print(f"无法读取文件: {full_path}, 原因: {e}")
+                    continue
+
+                out_f.write(f"=== File Path: {full_path} ===\n")
+                out_f.write(content)
+                out_f.write("\n\n")
+
+if __name__ == "__main__":
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    repo_dir = script_dir
+    output_txt = os.path.join(script_dir, "输出.txt")
+
+    # 这里我们专门排除 .svg, .png, .jpg
+    exclude_list = ['.svg', '.png', '.jpg', '.jpeg', '.gif']
+
+    collect_code_files_to_txt(
+        repo_path=repo_dir,
+        output_file=output_txt,
+        exclude_extensions=exclude_list,
+        encoding='utf-8'
+    )
+
+    print(f"文件收集完成，结果已保存到：{output_txt}")