Update app.py

app.py

@@ -1,7 +1,4 @@
-from dotenv import load_dotenv
 import streamlit as st
-import os
-import google.generativeai as genai
 from PIL import Image
 from PyPDF2 import PdfReader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -9,6 +6,9 @@ from langchain_google_genai import GoogleGenerativeAIEmbeddings, ChatGoogleGener
 from langchain.vectorstores import FAISS
 from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
+import os
+import google.generativeai as genai
+from dotenv import load_dotenv
 
 load_dotenv() # Load all env variables 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
@@ -76,13 +76,23 @@ def get_gemini_response(input,image):
 st.set_page_config(page_title='Combined Streamlit Application')
 st.header("Streamlit Application")
 
-# PDF Chat Section
-user_question = st.text_input("Ask a Question from the PDF Files")
-if user_question:
-    user_input(user_question)
+# Define the different applications
+applications = {
+    "PDF Chat": "pdf_chat",
+    "Image Chat": "image_chat",
+    "Q&A Chat": "qa_chat"
+}
+
+# Render the dropdown menu
+selected_app = st.sidebar.selectbox("Select Application", list(applications.keys()))
+
+# Function to display PDF Chat application
+def pdf_chat():
+    st.header("PDF Chat Application")
+    user_question = st.text_input("Ask a Question from the PDF Files")
+    if user_question:
+        user_input(user_question)
 
-with st.sidebar:
-    st.title("Menu:")
     pdf_docs = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
     if st.button("Submit & Process"):
         with st.spinner("Processing..."):
@@ -91,39 +101,51 @@ with st.sidebar:
             get_vector_store(text_chunks)
             st.success("Done")
 
-# Image Chat Section
-input_text = st.text_input("Input for Gemini Pro:", key="input_gemini")
-uploaded_file = st.file_uploader("Choose an image...", type="jpg")
-if uploaded_file is not None:
-    image = Image.open(uploaded_file)
-    st.image(image, caption="Uploaded Image", use_column_width=True)
-
-submit_gemini = st.button("Ask Gemini Pro")
-
-if submit_gemini:
-    response_gemini = get_gemini_response(input_text, image)
-    st.subheader("Gemini Pro Response:")
-    st.write(response_gemini)
-
-# Q&A Chat Section
-st.header("Q&A Chat Section")
-
-# Initialize session state for chat history if it doesn't exist
-if 'chat_history' not in st.session_state:
-    st.session_state['chat_history'] = []
-
-input_qa = st.text_area("Input for Q&A:", key="input_qa")
-submit_qa = st.button("Ask the question")
-
-if submit_qa and input_qa:
-    response_qa = get_gemini_response(input_qa)
-    # Add user query and response to session state chat history
-    st.session_state['chat_history'].append(("You", input_qa))
-    st.subheader("Q&A Response:")
-    for chunk in response_qa:
-        st.write(chunk.text)
-        st.session_state['chat_history'].append(("Gemini Pro", chunk.text))
-
-st.subheader("Q&A Chat History:")
-for role, text in st.session_state['chat_history']:
-    st.write(f"{role}: {text}")
+# Function to display Image Chat application
+def image_chat():
+    st.header("Image Chat Application")
+    input_text = st.text_input("Input for Gemini Pro:", key="input_gemini")
+    uploaded_file = st.file_uploader("Choose an image...", type="jpg")
+    if uploaded_file is not None:
+        image = Image.open(uploaded_file)
+        st.image(image, caption="Uploaded Image", use_column_width=True)
+
+    submit_gemini = st.button("Ask Gemini Pro")
+
+    if submit_gemini:
+        response_gemini = get_gemini_response(input_text, image)
+        st.subheader("Gemini Pro Response:")
+        st.write(response_gemini)
+
+# Function to display Q&A Chat application
+def qa_chat():
+    st.header("Q&A Chat Application")
+    # Initialize session state for chat history if it doesn't exist
+    if 'chat_history' not in st.session_state:
+        st.session_state['chat_history'] = []
+
+    input_qa = st.text_area("Input for Q&A:", key="input_qa")
+    submit_qa = st.button("Ask the question")
+
+    if submit_qa and input_qa:
+        response_qa = get_gemini_response(input_qa)
+        # Add user query and response to session state chat history
+        st.session_state['chat_history'].append(("You", input_qa))
+        st.subheader("Q&A Response:")
+        for chunk in response_qa:
+            st.write(chunk.text)
+            st.session_state['chat_history'].append(("Gemini Pro", chunk.text))
+
+    st.subheader("Q&A Chat History:")
+    for role, text in st.session_state['chat_history']:
+        st.write(f"{role}: {text}")
+
+# Map selected application to corresponding function
+selected_app_func = {
+    "PDF Chat": pdf_chat,
+    "Image Chat": image_chat,
+    "Q&A Chat": qa_chat
+}
+
+# Run the selected application function
+selected_app_func[selected_app]()