Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,66 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from transformers import pipeline
+from PIL import Image
+import torch
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# --- Config ---
+st.set_page_config(page_title="Image → Prompt", page_icon="🖼️", layout="centered")
+
+# Laad model (gecached)
+@st.cache_resource
+def load_model():
+    return pipeline(
+        "image-to-text",
+        model="llava-hf/llava-v1.6-34b-hf",
+        torch_dtype=torch.float16,
+        device_map="auto"
+    )
+
+pipe = load_model()
+
+# Stijlen
+STIJLEN = {
+    "Realistisch": ", fotorealistisch, 8k, scherp, natuurlijk licht",
+    "Anime": ", anime stijl, gedetailleerde ogen, Studio Ghibli, 4k",
+    "Olieverf": ", olieverf op doek, impressionistisch, Van Gogh stijl",
+    "Cyberpunk": ", cyberpunk, neon lichten, regenachtige straten, Blade Runner",
+    "Geen stijl": ""
+}
+
+# --- UI ---
+st.title("🖼️ Image → Prompt Generator")
+st.markdown("*Upload een foto, kies een stijl, krijg een perfecte AI-prompt.*")
+
+col1, col2 = st.columns([1, 1])
+with col1:
+    uploaded_file = st.file_uploader("Kies een afbeelding", type=["png", "jpg", "jpeg"])
+with col2:
+    stijl = st.selectbox("Kies een stijl", options=list(STIJLEN.keys()))
+
+if uploaded_file:
+    image = Image.open(uploaded_file).convert("RGB")
+    st.image(image, caption="Jouw afbeelding", use_column_width=True)
+
+    with st.spinner("Prompt genereren..."):
+        result = pipe(image)
+        raw_prompt = result[0]["generated_text"]
+
+        # Optimaliseer
+        final_prompt = (
+            raw_prompt
+            .replace("a photo of", "")
+            .replace("an image of", "")
+            .strip()
+            .capitalize()
+        )
+        final_prompt += STIJLEN[stijl]
+
+    st.success("**Prompt:**")
+    st.code(final_prompt, language=None)
+
+    st.download_button(
+        "📥 Download prompt",
+        final_prompt,
+        file_name="prompt.txt",
+        mime="text/plain"
+    )