Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
 import google.generativeai as genai
-from PIL import Image
 import os
-import time
 
 # Configure the API key
 api_key = 'AIzaSyC13hxiLDZQagfXybLKCklbTlBDsO4vHwQ'
@@ -13,21 +11,12 @@ def process_image(image_path, user_question=None):
     try:
         # Upload the image to Google Generative AI
         sample_file = genai.upload_file(path=image_path, display_name="Sample drawing")
-        #st.write(f"Uploaded file '{sample_file.display_name}' as: {sample_file.uri}")
-
-        # Retrieve the uploaded file
-        file = genai.get_file(name=sample_file.name)
-        #st.write(f"Retrieved file '{file.display_name}' as: {sample_file.uri}")
 
         # Use the Generative AI model to analyze the image
         model = genai.GenerativeModel(model_name="models/gemini-2.0-flash-exp")
 
         # Prepare the prompt
-        if user_question:
-            prompt = user_question
-        else:
-            prompt = "Explain the scene in the image in detail."
-
+        prompt = user_question if user_question else "Explain the scene in the image in detail."
         response = model.generate_content([prompt, sample_file])
 
         # Display the response
@@ -36,7 +25,6 @@ def process_image(image_path, user_question=None):
 
         # Delete the uploaded file
         genai.delete_file(sample_file.name)
-        #st.write(f"Deleted {sample_file.display_name}.")
 
     except Exception as e:
         st.error(f"An error occurred: {e}")
@@ -60,11 +48,18 @@ def set_cyberpunk_style():
         }
 
         .stButton button {
-            background-color: #ff00ff;
-            color: #0a0a0a;
-            border: 2px solid #00ff00;
+            background-color: #000000 !important;
+            color: #00ff00 !important;
+            border: 2px solid #00ff00 !important;
             font-family: 'Orbitron', sans-serif;
             font-weight: bold;
+            transition: all 0.3s ease;
+        }
+
+        .stButton button:hover {
+            background-color: #00ff00 !important;
+            color: #000000 !important;
+            border: 2px solid #000000 !important;
         }
 
         .stTextInput input {
@@ -101,7 +96,7 @@ def main():
     set_cyberpunk_style()
 
     # App title and description
-    st.title("VisioX: Cyberpunk Vision")
+    st.title("VisioX : SXR")
     st.markdown(
         """
         <div style="color: #00ff00;">
@@ -124,7 +119,7 @@ def main():
 
         # Display the captured image
         st.markdown("### Captured Image")
-        st.image(image_path, caption="Your Moment",)
+        st.image(image_path, caption="Your Moment")
 
         # Add a text input for user questions
         st.markdown("### Ask a Question")
@@ -141,7 +136,6 @@ def main():
 
         # Clean up the temporary file
         os.remove(image_path)
-        #st.write("Temporary image file deleted.")
 
 # Run the app
 if __name__ == "__main__":