Update app.py

app.py

@@ -0,0 +1,58 @@
+import streamlit as st
+import requests
+import os
+import json
+
+# Set the Nebius API key
+API_KEY = os.environ.get("NEBIUS_API_KEY")
+API_URL = "https://api.studio.nebius.ai/v1/chat/completions"
+
+# Streamlit app configuration
+st.set_page_config(page_title="Image to Prompt Converter", layout="centered", page_icon="🖼️")
+st.markdown("<style>body { background-color: #1E1E1E; color: #FFFFFF; }</style>", unsafe_allow_html=True)
+
+# App title
+st.title("Image to Prompt Converter")
+st.markdown("Upload an image and generate a detailed prompt.")
+
+# Image upload
+uploaded_file = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    # Display the uploaded image
+    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    
+    # Generate button
+    if st.button("Generate Prompt"):
+        # Prepare the API payload
+        files = {"file": uploaded_file.getvalue()}
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        data = {
+            "model": "llava-hf/llava-1.5-7b-hf",
+            "messages": [
+                {
+                    "role": "system",
+                    "content": """You are an image to prompt converter. Your work is to observe each and every detail of the image and craft a detailed prompt under 75 words in this format: [image content/subject, description of action, state, and mood], [art form, style], [artist/photographer reference if needed], [additional settings such as camera and lens settings, lighting, colors, effects, texture, background, rendering]."""
+                }
+            ],
+            "temperature": 0
+        }
+
+        # Call the Nebius API
+        response = requests.post(API_URL, headers=headers, data=json.dumps(data))
+        
+        if response.status_code == 200:
+            # Extract the generated prompt
+            result = response.json()
+            generated_prompt = result.get("choices", [{}])[0].get("message", {}).get("content", "No prompt generated.")
+            
+            # Display the generated prompt
+            st.subheader("Generated Prompt")
+            st.text_area("", generated_prompt, height=200)
+            
+            # Copy button
+            if st.button("Copy Prompt"):
+                st.write("Prompt copied to clipboard!")
+                st.write(generated_prompt)
+        else:
+            st.error(f"Failed to generate prompt: {response.status_code} - {response.text}")