Update app.py

app.py

@@ -0,0 +1,120 @@
+import streamlit as st
+import google.generativeai as genai
+from google.generativeai import types
+from PIL import Image
+import io
+import os # Import os for environment variable check (useful for local testing)
+
+# --- Configuration ---
+
+# Configure the Gemini API key.
+# Streamlit Secrets is the recommended way for Hugging Face Spaces.
+# Add your API key to `.streamlit/secrets.toml` like this:
+# GEMINI_API_KEY = "YOUR_API_KEY_HERE"
+# For local testing, you might also set it as an environment variable.
+try:
+    # Prefer Streamlit secrets if available
+    if "GEMINI_API_KEY" in st.secrets:
+        genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
+        st.success("Gemini API key loaded from secrets.")
+    elif os.getenv("GEMINI_API_KEY"):
+         # Fallback to environment variable (useful for local dev without secrets)
+         genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+         st.warning("Gemini API key loaded from environment variable. Use Streamlit Secrets for deployment.")
+    else:
+        st.error("GEMINI_API_KEY not found in Streamlit secrets or environment variables.")
+        st.stop() # Stop execution if API key is missing
+
+except Exception as e:
+    st.error(f"Failed to configure Gemini API: {e}")
+    st.stop()
+
+# Define the model name
+MODEL_NAME = "gemini-2.0-flash-exp" # Using the model from your code
+
+# --- Streamlit UI ---
+
+st.title("🎨 Gemini Native Image Generation")
+st.markdown("Enter a detailed prompt below to generate an image using the Gemini API.")
+
+# Input area for the prompt
+prompt = st.text_area(
+    "Enter your image prompt:",
+    height=150,
+    placeholder="e.g., A spaceship landing on a alien planet, digital art"
+)
+
+# Button to trigger generation
+generate_button = st.button("Generate Image")
+
+# --- Generation Logic ---
+
+if generate_button and prompt:
+    # Show a spinner while generating
+    with st.spinner("Generating image... Please wait."):
+        try:
+            # Call the Gemini API
+            response = genai.generate_content(
+                model=MODEL_NAME,
+                # Pass the prompt as a list within contents
+                contents=[prompt],
+                config=types.GenerateContentConfig(
+                    response_modalities=["Image"] # Only request Image modality
+                ),
+            )
+
+            # Check if the response has candidates and content parts
+            if response.candidates and response.candidates[0].content.parts:
+                # Find the image part (inline_data)
+                img_part = None
+                for part in response.candidates[0].content.parts:
+                    if part.inline_data:
+                        img_part = part.inline_data
+                        break # Found the image part
+
+                if img_part:
+                    img_data = img_part.data # This is the bytes data
+                    mime_type = img_part.mime_type # This is the mime type (e.g., 'image/png')
+
+                    # Display the image using Streamlit's st.image
+                    try:
+                        # Use PIL to open the image from bytes data
+                        image = Image.open(io.BytesIO(img_data))
+                        st.image(image, caption="Generated Image", use_column_width=True)
+                        st.success("Image generated successfully!")
+
+                        # Optional: Add a download button
+                        st.download_button(
+                            label="Download Image",
+                            data=img_data,
+                            file_name="generated_image.png", # You might want a more dynamic name
+                            mime=mime_type
+                        )
+
+                    except Exception as e:
+                        st.error(f"Error processing or displaying image data: {e}")
+                        # You could print raw data info for debugging:
+                        # st.write(f"Raw data size: {len(img_data)} bytes, MIME: {mime_type}")
+
+
+                else:
+                    st.warning("Generation successful, but no image data was returned for this prompt.")
+                    # Check for text response if image wasn't found (though we requested only Image)
+                    text_parts = [p.text for p in response.candidates[0].content.parts if p.text]
+                    if text_parts:
+                        st.write("Text response (no image found):")
+                        for text in text_parts:
+                            st.write(text)
+
+            else:
+                st.error("No valid response received from the API.")
+                # Print response for debugging if needed
+                # st.write(response)
+
+
+        except Exception as e:
+            st.error(f"An API error occurred during generation: {e}")
+            # Often API errors include details in the exception object
+
+elif generate_button and not prompt:
+    st.warning("Please enter a prompt to generate an image.")