mirror of
https://github.com/geoffsee/open-gsio.git
synced 2025-09-08 22:56:46 +00:00
Remove deprecated image generation and rate limiting services
This commit removes the image-generation-service and rate-limiter workers along with their corresponding configurations, scripts, and dependencies. These services are no longer required, streamlining the project's codebase and reducing maintenance overhead.
This commit is contained in:
geoffsee
committed by
Geoff Seemueller
Geoff Seemueller
parent
fb41f5f816
commit
2ed0c5e280
@@ -1,114 +0,0 @@
|
||||
import { OpenAI } from "openai";
|
||||
import { WorkerEntrypoint } from "cloudflare:workers";
|
||||
import Replicate from "replicate";
|
||||
|
||||
export default class extends WorkerEntrypoint {
|
||||
strategy;
|
||||
|
||||
constructor(ctx, env) {
|
||||
super(ctx, env);
|
||||
switch (env.TEXT2IMAGE_PROVIDER) {
|
||||
case "replicate":
|
||||
this.strategy = new ReplicateStrategy(env);
|
||||
break;
|
||||
case "openai":
|
||||
this.strategy = new OpenAiStrategy(env);
|
||||
break;
|
||||
default:
|
||||
throw "Invalid or missing image provider";
|
||||
}
|
||||
}
|
||||
|
||||
async fetch(request) {
|
||||
const { pathname, searchParams } = new URL(request.url);
|
||||
|
||||
if (pathname === "/generate") {
|
||||
const prompt =
|
||||
searchParams.get("prompt") || "A futuristic city with flying cars";
|
||||
const size = searchParams.get("size") || "1024x1024";
|
||||
|
||||
try {
|
||||
const imageData = await this.strategy.generateImage(prompt, size);
|
||||
if (isURL(imageData)) {
|
||||
return handleUrlResponse(imageData);
|
||||
} else {
|
||||
// Directly return image data as in ReplicateStrategy
|
||||
return handleImageDataResponse(imageData);
|
||||
}
|
||||
} catch (error) {
|
||||
console.error(error);
|
||||
return new Response("Image generation failed.", { status: 500 });
|
||||
}
|
||||
}
|
||||
|
||||
return new Response("Not Found", { status: 404 });
|
||||
}
|
||||
}
|
||||
|
||||
class OpenAiStrategy {
|
||||
constructor(env) {
|
||||
this.env = env;
|
||||
}
|
||||
|
||||
async generateImage(prompt, size) {
|
||||
this.openai = new OpenAI({ apiKey: env.OPENAI_API_KEY });
|
||||
const response = await this.openai.images.generate({
|
||||
model: this.env.IMAGE_MODEL,
|
||||
prompt,
|
||||
n: 1,
|
||||
response_format: "url",
|
||||
});
|
||||
return response.data[0].url;
|
||||
}
|
||||
}
|
||||
|
||||
class ReplicateStrategy {
|
||||
constructor(env) {
|
||||
this.env = env;
|
||||
}
|
||||
|
||||
async generateImage(prompt, size) {
|
||||
const replicate = new Replicate({ auth: this.env.REPLICATE_TOKEN });
|
||||
const output = await replicate.run(this.env.IMAGE_MODEL, {
|
||||
input: {
|
||||
prompt,
|
||||
aspect_ratio: "1:1",
|
||||
output_format: "webp",
|
||||
output_quality: 100,
|
||||
safety_tolerance: 2,
|
||||
height: parseInt(size.split("x").at(0).replace("x", "")),
|
||||
width: parseInt(size.split("x").at(1).replace("x", "")),
|
||||
prompt_upsampling: true,
|
||||
},
|
||||
});
|
||||
return output;
|
||||
}
|
||||
}
|
||||
|
||||
function isURL(imageUrl) {
|
||||
const urlPattern =
|
||||
/^(https?:\/\/)?([a-zA-Z0-9-]+\.)+[a-zA-Z]{2,}(:\d+)?(\/\S*)?$/;
|
||||
return urlPattern.test(imageUrl);
|
||||
}
|
||||
|
||||
async function handleUrlResponse(iUrl) {
|
||||
const imageResponse = await fetch(iUrl);
|
||||
if (!imageResponse.ok) {
|
||||
throw new Error(`Failed to fetch image: ${imageResponse.statusText}`);
|
||||
}
|
||||
const headers = new Headers(imageResponse.headers);
|
||||
headers.set("Content-Disposition", `inline; filename="generated_image.png"`);
|
||||
|
||||
return new Response(imageResponse.body, {
|
||||
headers,
|
||||
status: imageResponse.status,
|
||||
});
|
||||
}
|
||||
|
||||
async function handleImageDataResponse(imageData) {
|
||||
const headers = new Headers();
|
||||
headers.set("Content-Type", "image/png");
|
||||
headers.set("Content-Disposition", `inline; filename="generated_image.png"`);
|
||||
|
||||
return new Response(imageData, { headers, status: 200 });
|
||||
}
|
||||
@@ -1,16 +0,0 @@
|
||||
main="main.js"
|
||||
name = "image-generation-service"
|
||||
compatibility_date = "2024-12-20"
|
||||
node_compat = true
|
||||
dev.port = 3002
|
||||
|
||||
routes = [
|
||||
{ pattern = "text2image.seemueller.io", custom_domain = true }
|
||||
]
|
||||
|
||||
[vars]
|
||||
OPENAI_API_KEY = ""
|
||||
OPENAI_API_ENDPOINT = ""
|
||||
IMAGE_MODEL = "black-forest-labs/flux-1.1-pro"
|
||||
REPLICATE_TOKEN = ""
|
||||
TEXT2IMAGE_PROVIDER = "replicate"
|
||||
@@ -1,31 +0,0 @@
|
||||
interface Env {
|
||||
TEXT2IMAGE_RATE_LIMITER: any;
|
||||
}
|
||||
|
||||
export default {
|
||||
async fetch(request, env): Promise<Response> {
|
||||
const { pathname } = new URL(request.url);
|
||||
|
||||
const { success } = await env.TEXT2IMAGE_RATE_LIMITER.limit({
|
||||
key: pathname,
|
||||
});
|
||||
if (!success) {
|
||||
const svg = `
|
||||
<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 500 150" width="500" height="150">
|
||||
<rect width="100%" height="100%" fill="#f8d7da" />
|
||||
<text x="50%" y="50%" text-anchor="middle" dominant-baseline="middle" font-family="Arial, sans-serif" font-size="24" fill="#721c24">
|
||||
Sorry! Rate limit exceeded, try again in a couple minutes.
|
||||
</text>
|
||||
</svg>
|
||||
`;
|
||||
return new Response(svg, {
|
||||
status: 429,
|
||||
headers: {
|
||||
"Content-Type": "image/svg+xml",
|
||||
},
|
||||
});
|
||||
}
|
||||
|
||||
return new Response(`Success!`);
|
||||
},
|
||||
} satisfies ExportedHandler<Env>;
|
||||
@@ -1,22 +0,0 @@
|
||||
name = "rate-limiter"
|
||||
main = "index.ts"
|
||||
compatibility_date = "2024-12-20"
|
||||
node_compat = true
|
||||
dev.port = 3003
|
||||
|
||||
routes = [
|
||||
{ pattern = "text2image.seemueller.io/generate*", zone_name = "seemueller.io" },
|
||||
]
|
||||
|
||||
# The rate limiting API is in open beta.
|
||||
[[unsafe.bindings]]
|
||||
name = "TEXT2IMAGE_RATE_LIMITER"
|
||||
type = "ratelimit"
|
||||
# An identifier you define, that is unique to your Cloudflare account.
|
||||
# Must be an integer.
|
||||
namespace_id = "1001"
|
||||
|
||||
# Limit: the number of requests allowed within a given period in a single
|
||||
# Cloudflare location
|
||||
# Period: the duration of the period, in seconds. Must be either 10 or 60
|
||||
simple = { limit = 26, period = 101}
|
||||
Reference in New Issue
Block a user