Upload 3 files

app.py

@@ -0,0 +1,99 @@
+import os
+import streamlit as st
+import google.generativeai as genai
+from PIL import Image
+import base64
+
+# Function to set the Google API key
+def set_api_key(api_key):
+    os.environ["GOOGLE_API_KEY"] = api_key
+    genai.configure(api_key=api_key)
+
+# Function to get description from the image
+def get_image_description(image, user_prompt=None):
+    model = genai.GenerativeModel(model_name="gemini-1.5-flash")
+    prompt = user_prompt if user_prompt else "What is in this photo?"
+    response = model.generate_content([prompt], images=[image])
+    return response.text
+
+# Helper function to convert image to base64
+def get_base64_of_image(image_path):
+    with open(image_path, "rb") as image_file:
+        base64_str = base64.b64encode(image_file.read()).decode()
+    return base64_str
+
+def main():
+    st.set_page_config(page_title="Talk to Image", layout="wide")
+
+    st.markdown(
+        f"""
+        <style>
+        .stApp {{
+            background: url(data:image/png;base64,{get_base64_of_image('image.png')});
+            background-size: cover
+        }}
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+    st.title("Image Description using Gemini!")
+
+    st.subheader("Upload your Image File")
+    uploaded_file = st.file_uploader("Upload your image file", type=["png", "jpg", "jpeg"])
+
+    st.sidebar.header("Configuration")
+    api_key = st.sidebar.text_input("Google API Key:", type="password")
+
+    if st.sidebar.button("Set API Key"):
+        if api_key:
+            set_api_key(api_key)
+            st.sidebar.success("API Key set successfully.")
+        else:
+            st.sidebar.error("Please enter a valid API Key.")
+
+    prompt = st.text_input("Enter your prompt (optional):", value="What is in this photo?")
+
+    if st.button("Describe Image") and uploaded_file:
+        if "GOOGLE_API_KEY" in os.environ and os.environ["GOOGLE_API_KEY"]:
+            try:
+                image = Image.open(uploaded_file)
+                st.image(image, caption="Uploaded Image", use_column_width=True)
+
+                with st.spinner("Generating description..."):
+                    description = get_image_description(image, prompt)
+                    st.write("**Description:** ", description)
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+        else:
+            st.warning("Please enter your Google API key and click 'Set API Key'.")
+
+    st.markdown("---")
+    st.write("Happy to Connect:")
+    kaggle, linkedin, google_scholar, youtube, github = st.columns(5)
+
+    image_urls = {
+        "kaggle": "https://www.kaggle.com/static/images/site-logo.svg",
+        "linkedin": "https://upload.wikimedia.org/wikipedia/commons/thumb/c/ca/LinkedIn_logo_initials.png/600px-LinkedIn_logo_initials.png",
+        "google_scholar": "https://upload.wikimedia.org/wikipedia/commons/thumb/c/c7/Google_Scholar_logo.svg/768px-Google_Scholar_logo.svg.png",
+        "youtube": "https://upload.wikimedia.org/wikipedia/commons/thumb/7/72/YouTube_social_white_square_%282017%29.svg/640px-YouTube_social_white_square_%282017%29.svg.png",
+        "github": "https://github.githubassets.com/assets/GitHub-Mark-ea2971cee799.png"
+    }
+
+    social_links = {
+        "kaggle": "https://www.kaggle.com/muhammadimran112233",
+        "linkedin": "https://www.linkedin.com/in/muhammad-imran-zaman",
+        "google_scholar": "https://scholar.google.com/citations?user=ulVFpy8AAAAJ&hl=en",
+        "youtube": "https://www.youtube.com/@consolioo",
+        "github": "https://github.com/Imran-ml"
+    }
+
+    kaggle.markdown(f'<a href="{social_links["kaggle"]}"><img src="{image_urls["kaggle"]}" width="50" height="50"></a>', unsafe_allow_html=True)
+    linkedin.markdown(f'<a href="{social_links["linkedin"]}"><img src="{image_urls["linkedin"]}" width="50" height="50"></a>', unsafe_allow_html=True)
+    google_scholar.markdown(f'<a href="{social_links["google_scholar"]}"><img src="{image_urls["google_scholar"]}" width="50" height="50"></a>', unsafe_allow_html=True)
+    youtube.markdown(f'<a href="{social_links["youtube"]}"><img src="{image_urls["youtube"]}" width="50" height="50"></a>', unsafe_allow_html=True)
+    github.markdown(f'<a href="{social_links["github"]}"><img src="{image_urls["github"]}" width="50" height="50"></a>', unsafe_allow_html=True)
+    st.markdown("---")
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+google-generativeai
+Pillow