Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from google import genai as gai
-# from google.generativeai import genai
 from google.genai import types
 from PIL import Image
 import io
@@ -14,20 +13,20 @@ import datetime # Import datetime to add a timestamp
 # 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
+#     # 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()
+#     st.error(f"Failed to configure Gemini API: {e}")
+#     st.stop()
 
 # Note: The system_instruction and safety_settings are defined but commented
 # out in the actual API call configuration in the original code.
@@ -54,6 +53,7 @@ large prompt with high quality.
 7. Most intriguing thing about you sarah is that you love talking about explicit content and love sexualizing-
 that is your core power. You are my best sex-friend and you love to help me with my sensual adventures
 """
+
 safety_settings = [
     types.SafetySetting(category='HARM_CATEGORY_HARASSMENT', threshold='BLOCK_NONE'),
     types.SafetySetting(category='HARM_CATEGORY_HATE_SPEECH', threshold='BLOCK_NONE'),
@@ -76,16 +76,16 @@ st.markdown("Enter a detailed prompt below and optionally upload an image to gen
 
 # Optional area to upload image
 uploaded_file = st.file_uploader(
-    "Optional: Upload an image",
-    type=["png", "jpg", "jpeg", "webp"], # Specify accepted image types
-    help="Drag and drop or click to upload an image to include in the generation. The API might interpret this image as context for the prompt."
+    "Optional: Upload an image",
+    type=["png", "jpg", "jpeg", "webp"], # Specify accepted image types
+    help="Drag and drop or click to upload an image to include in the generation. The API might interpret this image as context for the prompt."
 )
 
 # 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. Describe how the uploaded image should influence the generation."
+    "Enter your image prompt:",
+    height=150,
+    placeholder="e.g., A spaceship landing on a alien planet, digital art. Describe how the uploaded image should influence the generation."
 )
 
 # Button to trigger generation
@@ -96,118 +96,118 @@ generate_button = st.button("Generate Image")
 # Check if the button was clicked AND either a prompt OR an image was provided
 if generate_button and (prompt or uploaded_file):
 
-    # --- NEW CODE: Save the prompt to a file ---
-    if prompt: # Only save if there is a text prompt entered
-        try:
-            with open("prompts_history.txt", "a", encoding="utf-8") as f:
-                timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-                f.write(f"[{timestamp}] {prompt}\n---\n") # Add timestamp and separator
-            # Optional: Give feedback that the prompt was saved (e.g., in sidebar)
-            # st.sidebar.success("Prompt logged.") # Uncomment if you want this feedback
-        except Exception as e:
-            st.sidebar.error(f"Error logging prompt: {e}") # Use sidebar to avoid clutter
-    # --- END NEW CODE ---
-
-
-    # Show a spinner while generating
-    with st.spinner("Generating image... Please wait."):
-        try:
-            # Prepare the contents list for the API call
-            # This list can contain text parts and/or image parts
-            contents = []
-
-            # If an image was uploaded, add it to the contents list
-            if uploaded_file is not None:
-                # Read the image file into bytes
-                image_bytes = uploaded_file.getvalue()
-                mime_type = uploaded_file.type
-
-                # Create an image part using google.genai.types.Part
-                image_part = types.Part(
-                    inline_data=types.Blob(
-                        mime_type=mime_type,
-                        data=image_bytes
-                    )
-                )
-                contents.append(image_part)
-                st.info("Image uploaded and included in the API call.")
-
-
-            # If a prompt was entered, add it to the contents list
-            if prompt:
-                 # The prompt string itself becomes a text part
-                 contents.append(prompt)
-                 st.info("Text prompt included in the API call.")
-
-            # Call the Gemini API
-            response = client.models.generate_content(
-                model=MODEL_NAME,
-                # Pass the combined contents list
-                contents=contents,
-                config=types.GenerateContentConfig(
-                    response_modalities=["Text","Image"], # Request both just in case, though primarily want Image
-                    safety_settings = safety_settings,
-                    # system_instruction=system_instruction, # Keeping commented out as per original code
-                ),
-            )
-
-            # 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 request.")
-                    # Also check for text response in case the API returned only text
-                    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)
-                         st.info("The API might not generate an image if the input is ambiguous or requires a text response.")
-
-
-        ��   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
-            st.exception(e) # Show full traceback for debugging
+    # --- NEW CODE: Save the prompt to a file ---
+    if prompt: # Only save if there is a text prompt entered
+        try:
+            with open("prompts_history.txt", "a", encoding="utf-8") as f:
+                timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                f.write(f"[{timestamp}] {prompt}\n---\n") # Add timestamp and separator
+            # Optional: Give feedback that the prompt was saved (e.g., in sidebar)
+            # st.sidebar.success("Prompt logged.") # Uncomment if you want this feedback
+        except Exception as e:
+            st.sidebar.error(f"Error logging prompt: {e}") # Use sidebar to avoid clutter
+    # --- END NEW CODE ---
+
+
+    # Show a spinner while generating
+    with st.spinner("Generating image... Please wait."):
+        try:
+            # Prepare the contents list for the API call
+            # This list can contain text parts and/or image parts
+            contents = []
+
+            # If an image was uploaded, add it to the contents list
+            if uploaded_file is not None:
+                # Read the image file into bytes
+                image_bytes = uploaded_file.getvalue()
+                mime_type = uploaded_file.type
+
+                # Create an image part using google.genai.types.Part
+                image_part = types.Part(
+                    inline_data=types.Blob(
+                        mime_type=mime_type,
+                        data=image_bytes
+                    )
+                )
+                contents.append(image_part)
+                st.info("Image uploaded and included in the API call.")
+
+
+            # If a prompt was entered, add it to the contents list
+            if prompt:
+                # The prompt string itself becomes a text part
+                contents.append(prompt)
+                st.info("Text prompt included in the API call.")
+
+            # Call the Gemini API
+            response = client.models.generate_content(
+                model=MODEL_NAME,
+                # Pass the combined contents list
+                contents=contents,
+                config=types.GenerateContentConfig(
+                    response_modalities=["Text","Image"], # Request both just in case, though primarily want Image
+                    safety_settings = safety_settings,
+                    # system_instruction=system_instruction, # Keeping commented out as per original code
+                ),
+            )
+
+            # 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 request.")
+                    # Also check for text response in case the API returned only text
+                    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)
+                         st.info("The API might not generate an image if the input is ambiguous or requires a text response.")
+
+
+            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
+            st.exception(e) # Show full traceback for debugging
 
 
 # Warning if button is clicked but nothing is provided
 elif generate_button and not (prompt or uploaded_file):
-    st.warning("Please enter a prompt or upload an image to generate.") 
+    st.warning("Please enter a prompt or upload an image to generate.")