Update app.py

app.py

@@ -9,7 +9,7 @@ def generate_response(user_message):
     try:
         # Use Gemini API to generate a response
         response = model.generate_content(user_message)
-
+        
         # Check if the response has text
         if response.text:
             return response.text
@@ -18,33 +18,33 @@ def generate_response(user_message):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
-# Initialize session state for conversation history
-if 'conversation' not in st.session_state:
-    st.session_state.conversation = []
-
+# Set up the Streamlit page
+st.set_page_config(page_title="Gemini QA System", page_icon="🤖")
 st.title("Gemini QA System")
 st.write("Ask a question and get an answer from Gemini AI.")
 
-# User input
-user_message = st.text_input("Enter your question here...")
-
-if st.button("Submit"):
-    if user_message:
-        response = generate_response(user_message)
-        st.session_state.conversation.append((user_message, response))
-
-# Display conversation history
-if st.session_state.conversation:
-    st.write("### Conversation History")
-    for i, (question, answer) in enumerate(st.session_state.conversation):
-        st.write(f"**Q{i+1}:** {question}")
-        st.write(f"**A{i+1}:** {answer}")
-
-# Examples in the sidebar
-st.sidebar.title("Examples")
-st.sidebar.write("Click on an example to use it:")
-examples = ["What is the capital of France?", "Explain quantum computing in simple terms."]
-for example in examples:
-    if st.sidebar.button(example):
-        response = generate_response(example)
-        st.session_state.conversation.append((example, response))
+# Create the input text area
+user_input = st.text_area("Enter your question here...", height=100)
+
+# Create a button to generate the response
+if st.button("Generate Response"):
+    if user_input:
+        with st.spinner("Generating response..."):
+            response = generate_response(user_input)
+        st.subheader("Gemini's Response:")
+        st.write(response)
+    else:
+        st.warning("Please enter a question.")
+
+# Add example questions
+st.subheader("Example Questions:")
+example_questions = [
+    "What is the capital of France?",
+    "Explain quantum computing in simple terms."
+]
+for question in example_questions:
+    if st.button(question):
+        with st.spinner("Generating response..."):
+            response = generate_response(question)
+        st.subheader("Gemini's Response:")
+        st.write(response)