| from g4f.Provider.hf_space import BlackForestLabsFlux1Dev, G4F, BlackForestLabsFlux1Schnell |
| import asyncio |
| from flask import Flask, request, jsonify, Response, render_template |
|
|
| app = Flask(__name__) |
|
|
|
|
| class IMG: |
| def __init__(self, prompt: str, width: int = 1024, height: int = 1024, guidance_scale: float = 3.5, seed: int = 0): |
| self.prompt = prompt |
| self.width = width |
| self.height = height |
| self.guidance_scale = guidance_scale |
| self.seed = seed |
| self.messages = [{"role": "user", "content": self.prompt}] |
|
|
| async def _run_async_generator(self, generator): |
| """Runs the async generator and extracts image URLs safely.""" |
| results = [] |
| try: |
| async for result in generator: |
| if hasattr(result, "images") and isinstance(result.images, list): |
| results.extend(result.images) |
| else: |
| results.append(str(result)) |
| except Exception as e: |
| print("Error processing response:", e) |
| return results |
|
|
| def _generate_images(self, provider_class, model): |
| """Generic method to fetch images from any provider.""" |
| async def main(): |
| try: |
| async for result in provider_class.create_async_generator( |
| model=model, messages=self.messages, |
| width=self.width, height=self.height, |
| guidance_scale=self.guidance_scale, seed=self.seed |
| ): |
| yield result |
| except Exception as e: |
| print(f"Error generating images from {model}:", e) |
| yield f"Error: {e}" |
|
|
| return asyncio.run(self._run_async_generator(main())) |
|
|
| def BlackForest(self,model="black-forest-labs-flux-1-dev"): |
| if model in BlackForestLabsFlux1Dev.get_models(): |
| pass |
| else: |
| model = "black-forest-labs-flux-1-dev" |
| return self._generate_images(BlackForestLabsFlux1Dev, model) |
|
|
| def FluxMidJourny(self,model="flux"): |
| if model in G4F.get_models(): |
| pass |
| else: |
| model = "flux" |
| return self._generate_images(G4F, model) |
|
|
| def BlackForestSchnell(self,model="black-forest-labs-flux-1-schnell"): |
| if model in BlackForestLabsFlux1Schnell.get_models(): |
| pass |
| else: |
| model = "black-forest-labs-flux-1-schnell" |
| return self._generate_images(BlackForestLabsFlux1Schnell, model) |
|
|
|
|
| @app.route("/generate/image", methods=["POST"]) |
| def generate_image(): |
| data = request.json |
| prompt = data.get("prompt") |
| model = data.get("model", "black-forest-labs-flux-1-dev") |
| width = data.get("width", 1024) |
| height = data.get("height", 1024) |
| guidance_scale = data.get("guidance_scale", 3.5) |
| seed = data.get("seed", 0) |
| provider = data.get("provider", "flux") |
|
|
| if not prompt: |
| return jsonify({"error": "prompt is required"}), 400 |
|
|
| img = IMG(prompt, width, height, guidance_scale, seed) |
|
|
|
|
| def GenerateImage(): |
| if provider == "blackforestlabs": |
| return img.BlackForest(model) |
| elif provider == "flux": |
| return img.FluxMidJourny(model) |
| elif provider == "blackforestlabs-schnell": |
| return img.BlackForestSchnell(model) |
|
|
| result = GenerateImage() |
| print(result) |
| return jsonify({"Result" : result}), 200 |
|
|
| @app.route("/providers", methods=["GET"]) |
| def get_providers(): |
| return jsonify({"providers": ["blackforestlabs", "flux", "blackforestlabs-schnell"]}), 200 |
|
|
| @app.route("/generate/image/model", methods=["POST"]) |
| def get_models(): |
| data = request.json |
| provider = data.get("provider", "blackforestlabs") |
|
|
| if provider == "blackforestlabs": |
| return jsonify({"models": BlackForestLabsFlux1Dev.get_models()}), 200 |
| elif provider == "flux": |
| return jsonify({"models": G4F.get_models()}), 200 |
| elif provider == "blackforestlabs-schnell": |
| return jsonify({"models": BlackForestLabsFlux1Schnell.get_models()}), 200 |
| return jsonify({"error": "provider not found"}), 404 |
|
|
| @app.route("/", methods=["GET"]) |
| def index(): |
| return render_template("index.html") |
|
|
| if __name__ == "__main__": |
| app.run(port=7860) |
