Update AskMe.py

AskMe.py

@@ -1,43 +1,23 @@
-import streamlit as st
-from dotenv import load_dotenv
-import os
-import google.generativeai as genai
-
-# Load environment variables
-load_dotenv()
-api_key = os.getenv("GOOGLE_API_KEY")
-genai.configure(api_key=api_key)
-
-def show():
-    st.title("Curious Mind")
-    
-    # Display header and input text
-    st.header("What's on your mind?")
-    user_input = st.text_area("Ask a question:", key="input", placeholder="Type your question here...")
-    
-    # Add a submit button
-    submit_button = st.button("Get an Answer")
-    
-    # Display a spinner while processing
-    with st.spinner('Thinking...'):
-        if submit_button:
-            if user_input:
-                try:
-                    # Initialize model and generate response
-                    model = genai.GenerativeModel('gemini-1.0-pro-latest')
-                    response = model.generate_content(user_input)
-                    
-                    # Display response
-                    st.subheader("Here's what I found:")
-                    st.write(response.text)
-                except Exception as e:
-                    st.error(f"Oops! Something went wrong: {e}")
-            else:
-                st.warning("Please enter a question to continue.")
-    
-    # Optionally, add more interactive elements
-    st.sidebar.header("Settings")
-    st.sidebar.text("Adjust model parameters or choose a different model.")
-
-if __name__ == "__main__":
-    show()
+import streamlit as st
+from dotenv import load_dotenv
+load_dotenv()
+import os
+import google.generativeai as genai
+
+os.getenv("GOOGLE_API_KEY")
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+def show():    
+    st.header("QA Gemeni")
+    input=st.text_input("What's on your mind",key="input")
+
+    submit=st.button("Submit")
+
+    if submit:
+        model=genai.GenerativeModel('gemini-1.5-pro-latest')
+        st.subheader("The Response is")
+        if input!="":
+            response=model.generate_content(input)
+            st.write(response.text)
+
+        else:
+            st.write("Ask a Question First!!")