Update app.py

app.py

@@ -64,7 +64,7 @@ def analyze_image(image, prompt):
         if "<OD>" in predictions:
             for i, (bbox, label) in enumerate(zip(predictions["<OD>"]["bboxes"], predictions["<OD>"]["labels"])):
                 # Erweitere Filter für Kerzen
-                if "candle" not in label.lower() and "bar" not in label.lower() and "chart" not in label.lower():
+                if "candle" not in label.lower() and "bar" not in label.lower() and "chart" not in label.lower() and "candlestick" not in label.lower():
                     continue
                 xmin, ymin, xmax, ymax = map(int, bbox)
                 
@@ -106,4 +106,18 @@ def analyze_image(image, prompt):
         return {"prompt": prompt, "detections": detections}
 
     else:
-        return {"prompt": prompt, "description": "Unsupported prompt
+        return {"prompt": prompt, "description": "Unsupported prompt. Use 'Was siehst du auf dem Bild?' or 'List last 8 candles with their colors'."}
+
+# Erstelle Gradio-Schnittstelle
+iface = gr.Interface(
+    fn=analyze_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload an Image"),
+        gr.Textbox(label="Prompt", placeholder="Enter your prompt, e.g., 'Was siehst du auf dem Bild?' or 'List last 8 candles with their colors'")
+    ],
+    outputs="json",
+    title="Image Analysis with Florence-2-base",
+    description="Upload an image to analyze candlesticks or get a general description."
+)
+
+iface.launch()