Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -46,21 +46,22 @@ def generate_image(prompt: str, steps: int = 10, n_images: int = 1) -> list[Imag
|
|
46 |
if not prompt:
|
47 |
return "Bitte geben Sie einen Prompt ein."
|
48 |
|
|
|
|
|
|
|
|
|
|
|
49 |
try:
|
50 |
-
# KORRIGIERTER API-Aufruf-Methode: client.images.generate()
|
51 |
logger.info(f"Versuche, Bildgenerierung mit Modell {FLUX_MODEL} zu starten über client.images.generate()...")
|
52 |
-
response = client.images.generate(
|
53 |
model=FLUX_MODEL,
|
54 |
prompt=prompt,
|
55 |
-
steps=steps,
|
56 |
n=n_images
|
57 |
)
|
58 |
logger.info("Bildgenerierung erfolgreich angefordert.")
|
59 |
|
60 |
images = []
|
61 |
-
# Der Response-Typ von client.images.generate() ist möglicherweise anders als bei create().
|
62 |
-
# Wir müssen sicherstellen, dass wir auf die 'data' zugreifen können.
|
63 |
-
# Prüfen Sie die Struktur von 'response' bei Erfolg in den Logs, falls es immer noch Probleme gibt.
|
64 |
if hasattr(response, 'data') and response.data:
|
65 |
for image_data in response.data:
|
66 |
img_bytes = base64.b64decode(image_data.b64_json)
|
@@ -72,10 +73,10 @@ def generate_image(prompt: str, steps: int = 10, n_images: int = 1) -> list[Imag
|
|
72 |
return "Fehler: Ungültige Antwort von Together.ai. Bitte versuchen Sie es später erneut oder überprüfen Sie Ihr Guthaben."
|
73 |
|
74 |
except Exception as e:
|
75 |
-
logger.error(f"FEHLER bei der Bildgenerierung: {e}", exc_info=True)
|
76 |
return f"Fehler bei der Bildgenerierung: {e}. Eventuell API-Limit erreicht, Prompt ungültig oder Together.ai Problem."
|
77 |
|
78 |
-
# --- Gradio Interface Definition
|
79 |
with gr.Blocks() as demo:
|
80 |
gr.Markdown(
|
81 |
"""
|
@@ -93,11 +94,11 @@ with gr.Blocks() as demo:
|
|
93 |
)
|
94 |
steps_slider = gr.Slider(
|
95 |
minimum=1,
|
96 |
-
maximum=
|
97 |
-
value=
|
98 |
step=1,
|
99 |
label="Schritte (Steps)",
|
100 |
-
info="
|
101 |
)
|
102 |
num_images_slider = gr.Slider(
|
103 |
minimum=1,
|
|
|
46 |
if not prompt:
|
47 |
return "Bitte geben Sie einen Prompt ein."
|
48 |
|
49 |
+
# Zusätzliche Validierung für Schritte, falls der Slider-Wert das API-Limit überschreitet
|
50 |
+
if not (1 <= steps <= 4):
|
51 |
+
logger.error(f"Fehler: Schritte ({steps}) müssen zwischen 1 und 4 liegen.")
|
52 |
+
return "Fehler: Die Anzahl der Schritte muss zwischen 1 und 4 liegen."
|
53 |
+
|
54 |
try:
|
|
|
55 |
logger.info(f"Versuche, Bildgenerierung mit Modell {FLUX_MODEL} zu starten über client.images.generate()...")
|
56 |
+
response = client.images.generate(
|
57 |
model=FLUX_MODEL,
|
58 |
prompt=prompt,
|
59 |
+
steps=steps, # Dieser Wert kommt vom Slider
|
60 |
n=n_images
|
61 |
)
|
62 |
logger.info("Bildgenerierung erfolgreich angefordert.")
|
63 |
|
64 |
images = []
|
|
|
|
|
|
|
65 |
if hasattr(response, 'data') and response.data:
|
66 |
for image_data in response.data:
|
67 |
img_bytes = base64.b64decode(image_data.b64_json)
|
|
|
73 |
return "Fehler: Ungültige Antwort von Together.ai. Bitte versuchen Sie es später erneut oder überprüfen Sie Ihr Guthaben."
|
74 |
|
75 |
except Exception as e:
|
76 |
+
logger.error(f"FEHLER bei der Bildgenerierung: {e}", exc_info=True)
|
77 |
return f"Fehler bei der Bildgenerierung: {e}. Eventuell API-Limit erreicht, Prompt ungültig oder Together.ai Problem."
|
78 |
|
79 |
+
# --- Gradio Interface Definition ---
|
80 |
with gr.Blocks() as demo:
|
81 |
gr.Markdown(
|
82 |
"""
|
|
|
94 |
)
|
95 |
steps_slider = gr.Slider(
|
96 |
minimum=1,
|
97 |
+
maximum=4, # <--- DIESE ZEILE WURDE GEÄNDERT!
|
98 |
+
value=4, # <--- DIESE ZEILE WURDE GEÄNDERT!
|
99 |
step=1,
|
100 |
label="Schritte (Steps)",
|
101 |
+
info="Die Anzahl der Schritte muss zwischen 1 und 4 liegen für FLUX.1 [schnell]."
|
102 |
)
|
103 |
num_images_slider = gr.Slider(
|
104 |
minimum=1,
|