Update app.py

app.py

@@ -50,13 +50,10 @@ def retrieve_context(query, embed_model, index, df, k=3):
 # Ask the Groq LLM
 def ask_groq(query, context):
     prompt = f"""You are a helpful assistant. Use the context to answer the question.
-
 Context:
 {context}
-
 Question:
 {query}
-
 Answer:"""
 
     response = client.chat.completions.create(
@@ -74,11 +71,11 @@ st.subheader("🔍 Ask your question")
 sample_queries = df['query'].dropna().unique().tolist()
 col1, col2 = st.columns([3, 1])
 with col1:
-    query = st.text_input("Enter your question here:")
+    query = st.text_input("Enter your question here:", value=st.session_state.get("query", ""))
 with col2:
     if st.button("🎲 Random Sample"):
-        query = random.choice(sample_queries)
-        st.experimental_rerun()
+        st.session_state["query"] = random.choice(sample_queries)
+        st.rerun()
 
 # Handle query
 if query: