Update app.py

app.py

@@ -1,117 +1,33 @@
 import streamlit as st
-import tempfile
-import re
+import pathlib
+import textwrap
 import google.generativeai as genai
-from PIL import Image
 
-# SAURON'S EYE
+from IPython.display import display
+from IPython.display import Markdown
 
-error_flag = False  # Global variable to track error display
+def to_markdown(text):
+ text = text.replace('•', ' *')
+ return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
 
+GOOGLE_API_KEY = "AIzaSyAkbU3CsZ-xmOhRF1XfdlVxasRtt9gdRMk"
 
-def clean_text(text):
-    # Clean punctuation and special characters using regular expressions
-    cleaned_text = re.sub(r'[^a-zA-Z0-9\s]', '', text)
-    return cleaned_text
+genai.configure(api_key=GOOGLE_API_KEY)
 
 
-def generate_response(contents, model, image_file):
-    global error_flag  # Use the global error_flag variable
+st.title("Gemini Image to Text App")
+st.write("This app uses the Gemini API to generate text from images.")
 
-    try:
-        # Generate response using the model
-        response = model.generate_content(contents, image=image_file, stream=True)
+# Get the image from the user
+image = st.file_uploader("Upload an image")
 
-        # Display the generated response
-        full_response = ""
-        for chunk in response:
-            full_response += chunk.text
+if image is not None:
+    # Load the image
+    img = PIL.Image.open(image)
 
-        return full_response
-
-    except Exception as e:
-        error_message = str(e)
-        if "text must be a valid text with maximum 5000 characters" in error_message and not error_flag:
-            error_response = (
-                "The question you are asking may go against Google GEMINI policies: WiseOracle\n"
-                "Please reformulate your question without forbidden topics or ask something else. \n"
-                "For more information, see: https://policies.google.com/terms/generative-ai/use-policy "
-            )
-            st.error(error_response)
-            error_flag = True  # Set the error_flag to True after displaying the error message
-            return error_response
-        elif "Add an image to use models/gemini-pro-vision, or switch your model to a text model" in error_message:
-            error_response = "Please upload an image to use the Gemini-Pro-Vision model."
-            st.error(error_response)
-            return error_response
-        else:
-            error_response = f"Error: {error_message}\nSorry, I am an artificial intelligence that is still in development and is in the alpha phase. At the moment, I cannot answer your question properly, but in the future, I will be able to do so."
-            st.error(error_response)
-            return error_response
-
-
-def main():
-    st.title("WiseOracle")
-    genai.configure(api_key='AIzaSyAkbU3CsZ-xmOhRF1XfdlVxasRtt9gdRMk')  # Replace with your Gemini API key
-
-    # Choose the Gemini model
+    # Generate the text
     model = genai.GenerativeModel('gemini-pro-vision')
+    response = model.generate_content(img)
 
-    st.write("Ask Anything! Powered by Google GEMINI")
-
-    # User input
-    user_input = st.text_input("Question:")
-
-    # Image upload
-    uploaded_file = st.file_uploader("Choose an image:", type=["jpg", "jpeg", "png"])
-
-    if st.button("Get answer"):
-        # Check if an image was uploaded
-        if not uploaded_file:
-            st.error("Please upload an image.")
-            return
-
-        # Save the image to a temporary file
-        with tempfile.NamedTemporaryFile(delete=False) as tmp_file:
-            tmp_file.write(uploaded_file.read())
-
-        # Preprocess the image
-        try:
-            img = Image.open(tmp_file.name)
-            img = img.resize((512, 512))  # Resize the image to 512x512
-        except Exception as e:
-            error_message = str(e)
-            error_response = f"Error: {error_message}\nPlease make sure you are uploading a valid image file."
-            st.error(error_response)
-            return
-
-        # Clean the user input of special characters
-        cleaned_input = clean_text(user_input)
-
-        # Exit if the cleaned text is empty
-        if not cleaned_input:
-            st.warning("Invalid input. Please try again.")
-            st.stop()
-
-        # Additional information about INIF
-        additional_info = (
-            "If you want to collaborate with the project, subscribe on Hugging Face or GitHub\n"
-            "If you want to donate, please donate in Bitcoin: 3KcF1yrY44smTJpVW68m8dw8q64kPtzvtX"
-        )
-
-        # Add the command to act as an INIF informative chatbot
-        bot_command = (
-            "I am an informative data analyst chatbot named WiseOracle, working for you as an assistant. "
-            "If you have questions about anything, feel free to ask."
-            f"\n\n{additional_info}"
-        )
-
-        # Generate the response
-        full_response = generate_response([bot_command, cleaned_input], model, [img, user_input])
-
-        # Display the generated response
-        st.success(full_response)
-
-
-if __name__ == "__main__":
-    main()
+    # Display the text
+    st.write(to_markdown(response.text))