vab42 commited on
Commit
8d67e7a
·
verified ·
1 Parent(s): 6dd7fe2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
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( # <-- DIESE ZEILE WURDE GEÄNDERT!
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) # exc_info zeigt den vollständigen Traceback
76
  return f"Fehler bei der Bildgenerierung: {e}. Eventuell API-Limit erreicht, Prompt ungültig oder Together.ai Problem."
77
 
78
- # --- Gradio Interface Definition (Bleibt unverändert) ---
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=50,
97
- value=10,
98
  step=1,
99
  label="Schritte (Steps)",
100
- info="Höhere Schritte können die Qualität leicht verbessern, erhöhen aber die Generierungszeit."
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,