Updated Chatbot

app.py

@@ -1,49 +1,65 @@
-
-from dotenv import load_dotenv
-
-load_dotenv()  # take environment variables from .env.
-
 import streamlit as st
 import os
-import pathlib
-import textwrap
-
+from dotenv import load_dotenv
 import google.generativeai as genai
 
-from IPython.display import display
-from IPython.display import Markdown
-
-
-os.getenv("GOOGLE_API_KEY")
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+# Load environment variables
+load_dotenv()
+API_KEY = os.getenv("GOOGLE_API_KEY")
 
-## Function to load OpenAI model and get respones
+# Configure Gemini API
+genai.configure(api_key=API_KEY)
 model = genai.GenerativeModel('gemini-pro')
-chat = model.start_chat(history=[])
-def get_gemini_response(question):
-    
-    response =chat.send_message(question,stream=True)
-    return response
-
-##initialize our streamlit app
-
-st.set_page_config(page_title="Q&A Demo")
-
-st.header("Gemini Application")
 
-input=st.text_input("Input: ",key="input")
+# Streamlit Page Config
+st.set_page_config(page_title="Gemini Q&A App", layout="wide")
 
+# Initialize Chat History in Session State
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
 
-submit=st.button("Ask the question")
-
-## If ask button is clicked
-
-if submit:
-    
-    response=get_gemini_response(input)
-    st.subheader("The Response is")
-    for chunk in response:
-        print(st.write(chunk.text))
-        print("_"*80)
-    
-    st.write(chat.history)
+# Function to get response from Gemini API
+def get_gemini_response(question):
+    chat = model.start_chat(history=st.session_state.chat_history)
+    try:
+        response = chat.send_message(question, stream=True)
+        full_response = ""
+        for chunk in response:
+            full_response += chunk.text + " "
+        st.session_state.chat_history.append({"user": question, "bot": full_response})
+        return full_response
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# UI Layout
+st.title("🤖 Gemini AI - Interactive Q&A")
+st.write("Ask me anything and I'll try to answer!")
+
+# Sidebar for Settings
+with st.sidebar:
+    st.header("Settings")
+    if st.button("Clear Chat History"):
+        st.session_state.chat_history = []
+        st.success("Chat history cleared!")
+
+# User Input
+user_input = st.text_input("Your Question:", placeholder="Type your question here...")
+submit = st.button("Ask Gemini")
+
+# Response Handling
+if submit and user_input:
+    if not API_KEY:
+        st.error("API Key is missing. Please check your .env file.")
+    else:
+        with st.spinner("Thinking..."):
+            response = get_gemini_response(user_input)
+            st.subheader("Response:")
+            st.markdown(response)  # Render response in Markdown format
+
+# Display Chat History
+if st.session_state.chat_history:
+    st.subheader("Chat History")
+    for chat in st.session_state.chat_history:
+        st.markdown(f"**You:** {chat['user']}")
+        st.markdown(f"**Gemini:** {chat['bot']}")
+        st.write("---")