first commit

.gitattributes

@@ -0,0 +1,2 @@
+*.pdf filter=lfs diff=lfs merge=lfs -text
+*.PNG filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.streamlit/secrets.toml

.streamlit/config.toml

@@ -0,0 +1,2 @@
+[theme]
+base="light"

app.py

@@ -0,0 +1,49 @@
+import streamlit as st
+from pypdf import PdfReader
+import os
+from gemini_call import GeminiQanA
+
+if __name__ == "__main__":
+    #setup env variables
+    os.environ['GOOGLE_API_KEY']=st.secrets['GOOGLE_API_KEY']
+    st.set_page_config(page_title=None, page_icon=None, layout="centered", initial_sidebar_state="expanded", menu_items=None)
+    # Streamlit UI
+    st.title("📄 AI PDF Chatbot")
+    st.sidebar.image("zega_logo.PNG",width=300)
+    text_from_pdf:str = ""
+    # File uploader
+    uploaded_file = st.file_uploader("Upload a PDF (Max 50 pages)", type=["pdf"])
+    if uploaded_file:
+        pdf_document = PdfReader(uploaded_file)
+        
+        for page_num, page in enumerate(pdf_document.pages):
+            text = page.extract_text()
+            
+            text_from_pdf += text.strip().replace("\n" ," ")
+        
+        chatbot = GeminiQanA(text_from_pdf)
+        st.success("PDF uploaded and processed!")
+    else:
+        text_from_pdf = ""
+        if "messages" in st.session_state:
+            st.session_state.messages = []
+
+    # Chat interface
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    question = st.text_input("Ask a question about the document:")
+    if st.button("Ask AI") and question:
+        st.session_state.messages.append({"role": "user", "content": question})
+        with st.spinner("Fetching response from assistant..."):
+            answer = chatbot.answer_question(question)
+        st.session_state.messages.append({"role": "assistant", "content": answer})
+        
+        with st.chat_message("user"):
+            st.markdown(question)
+        with st.chat_message("assistant"):
+            st.markdown(answer)

gemini_call.py

@@ -0,0 +1,72 @@
+import google.generativeai as genai
+import os
+
+class GeminiQanA:
+    def __init__(self,text:str=''):
+        """Initializes the Gemini question answerer by loading the model."""
+        self.api_key = os.getenv("GOOGLE_API_KEY")
+        genai.configure(api_key=self.api_key)
+        self.model = self._load_model(text)
+
+    def _load_model(self,text:str):
+        """Loads the generative AI model with a system instruction."""
+        final_prompt = f'''Task:
+            You are an advanced AI system designed to answer questions based on a provided document. Your goal is to analyze the given text thoroughly and provide the most accurate, well-structured, and insightful response.
+
+            Instructions:
+
+            1. Carefully Read and Analyze the Document:
+
+            -Take your time to process the entire document before answering.
+            -Identify key details, themes, facts, and context relevant to the question.
+
+            2. Understand the Question Fully:
+
+            -Determine what specific information is being asked.
+            -If multiple interpretations are possible, consider the most relevant one based on the document.
+
+            3. Formulate a Comprehensive Answer:
+
+            -Provide a clear, concise, and informative response.
+            -Support your answer with direct references or logical inferences from the document.
+            -Maintain a structured response format for clarity.
+
+            4. Ensure Accuracy and Depth:
+
+            -Avoid assumptions beyond the given document.
+            -If the document does not contain the necessary information, state that explicitly.
+            -Provide additional context if it enhances the response.
+
+            5. Maintain a Professional and Neutral Tone:
+
+            -Present information objectively without personal bias.
+            -Keep the response clear and reader-friendly.
+
+            Example Format:
+
+            Document Provided:
+            "The solar system consists of the Sun and the celestial objects bound to it by gravity, including eight planets, moons, asteroids, and comets. The Earth is the third planet from the Sun and supports life due to its atmosphere, water, and suitable climate."
+
+            User Question:
+            "Why is Earth suitable for life?"
+
+            AI Response:
+            "Earth is suitable for life primarily due to three key factors: its atmosphere, presence of water, and favorable climate. The atmosphere contains oxygen and protects against harmful radiation, water is essential for biological processes, and the climate remains within a range that supports diverse ecosystems. These conditions make Earth uniquely habitable compared to other planets in the solar system."
+            
+            Provided text for refference:
+            {text}
+            
+            Question:
+                    '''
+        
+        return genai.GenerativeModel("gemini-1.5-pro", system_instruction=final_prompt)
+    
+    def answer_question(self, question: str) -> str:
+        """Performs the API call to Gemini and returns the sentiment analysis response."""
+        response = self.model.generate_content(question)
+        return response.text
+
+if __name__ == "__main__":
+    analyzer = GeminiQanA()
+    response = analyzer.answer_question("Hello,how are you?")
+    print(response)

requirements.txt

@@ -0,0 +1,4 @@
+protobuf==5.29.3
+pypdf==5.3.0
+streamlit==1.42.0
+google-generativeai==0.8.4