paperanalysis.py

from dotenv import load_dotenv
load_dotenv()  # load all the environment variables

import streamlit as st
import os
import google.generativeai as genai
from PIL import Image
from googletrans import Translator

# Initialize Google API key for translation
translator = Translator()

# Configure Google API for OpenAI
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))

# Function to load Google Gemini Pro Vision API And get response
def get_gemini_response(image, prompt):
    model = genai.GenerativeModel('gemini-pro-vision')
    response = model.generate_content([prompt, image[0]])
    return response.text

# Function to set up image data
def input_image_setup(uploaded_file):
    # Check if a file has been uploaded
    if uploaded_file is not None:
        # Read the file into bytes
        bytes_data = uploaded_file.getvalue()

        image_parts = [
            {
                "mime_type": uploaded_file.type,  # Get the mime type of the uploaded file
                "data": bytes_data
            }
        ]
        return image_parts
    else:
        raise FileNotFoundError("No file uploaded")

# Initialize Streamlit app
st.set_page_config(page_title="Edu For All",page_icon=":apple:")
st.header("Question Paper analysis")

# Define input prompt
input_prompt = """
You are an expert in analyzing question paper where you need to see the question from the image
and analyze the paper and say about the topics present in question paper and breif information about the those topics, also provide 5-7 or more similar questions that are  related to the questions in the 
question paper and the questions
is below format

1. Topic 1 - similar question
2. Topic 2 - similar question
----
----
Finally you can also mention some informative website and make sure you always provide the geeksforgeeks,IBM technology and MIT opencourseware website to prepare  and also mention 
extra topics for deeper learning , some practical examples to get some experimental learning
make sure follow all the prompt
"""

# File upload
uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
image = None
if uploaded_file is not None:
    image = Image.open(uploaded_file)
    st.image(image, caption="Uploaded Image.", use_column_width=True)

# Language selection
selected_language = st.selectbox("Select Language:", ["English", "Telugu", "Hindi"])

# Submit button
submit = st.button("Start Analysis")

# If submit button is clicked
if submit:
    if image:
        image_data = input_image_setup(uploaded_file)
        response = get_gemini_response(image_data, input_prompt)

        # Translation based on selected language
        if selected_language != "English":
            translated_response = translator.translate(response, dest=selected_language.lower())
            st.subheader("Translated Response:")
            st.write(translated_response.text)
        else:
            st.subheader("The Response is:")
            st.write(response)