Update app.py

app.py

@@ -117,7 +117,6 @@ def image_chat():
         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
@@ -125,4 +124,28 @@ def qa_chat():
         st.session_state['chat_history'] = []
 
     input_qa = st.text_area("Input for Q&A:", key="input_qa")
-    submit_qa = st.button("
+    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]()
+