Update app.py

app.py

@@ -1,18 +1,17 @@
 import os
-import gradio as gr
 import faiss
 import pickle
 import numpy as np
+import streamlit as st
 from sentence_transformers import SentenceTransformer
 from groq import Groq
 
-# ==== Connect to Groq using environment variable ====
-#client = Groq(api_key=os.environ.get("api_key5"))
+# ==== Setup ====
+st.set_page_config(page_title="🪖 US Army RAG Assistant", layout="wide")
+
+# ==== Connect to Groq ====
 groq_api_key = os.environ.get("api_key5")
 client = Groq(api_key=groq_api_key)
-# Store API key in a constant
-# Set up the Groq API key (replace with your actual key or use an environment variable)
-#client = Groq(api_key="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")
 
 # ==== Load FAISS index and metadata ====
 index = faiss.read_index("army_qa_faiss.index")
@@ -55,7 +54,10 @@ Answer:"""
 
     return response.choices[0].message.content.strip()
 
-# ==== Chat Interface ====
+# ==== Streamlit UI ====
+st.title("🪖 US Army RAG Assistant")
+st.markdown("Ask tactical questions based on **US Army Field Manual** knowledge.")
+st.divider()
 
 recommended_questions = [
     "What are the fundamentals of offensive operations?",
@@ -65,28 +67,29 @@ recommended_questions = [
     "Explain the role of reconnaissance in the offense."
 ]
 
-def chat(user_input, history):
-    if user_input.strip().lower() == "terminate":
-        return history + [[user_input, "👋 Goodbye, soldier. Stay sharp!"]], gr.update(interactive=False)
-    answer = rag_query(user_input)
-    return history + [[user_input, answer]], history
-
-with gr.Blocks() as demo:
-    gr.Markdown("### 🪖 US Army RAG Assistant\nAsk tactical questions based on FM manual knowledge.")
-    chatbot = gr.Chatbot()
-    textbox = gr.Textbox(label="Your question", placeholder="Type here (or type 'terminate' to exit)")
-    state = gr.State([])
-    submit = gr.Button("Submit")
-    clear = gr.Button("Clear")
+st.markdown("💡 **Recommended Questions:**")
+for q in recommended_questions:
+    st.markdown(f"- {q}")
 
-    with gr.Row():
-        gr.Markdown("##### 💡 Recommended Questions:")
-        for q in recommended_questions:
-            gr.Markdown(f"- {q}")
+# Chat state
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
 
-    submit.click(chat, [textbox, state], [chatbot, state])
-    clear.click(lambda: ([], []), None, [chatbot, state])
-
-if  __name__ == "__main__":
-  demo.launch()
+# ==== Input ====
+user_input = st.text_input("Your question", placeholder="Type your question here...")
 
+if st.button("Submit") and user_input:
+    if user_input.strip().lower() == "terminate":
+        st.session_state.chat_history.append((user_input, "👋 Goodbye, soldier. Stay sharp!"))
+    else:
+        answer = rag_query(user_input)
+        st.session_state.chat_history.append((user_input, answer))
+
+if st.button("Clear Chat"):
+    st.session_state.chat_history = []
+
+# ==== Display Chat ====
+for i, (user, bot) in enumerate(st.session_state.chat_history):
+    st.markdown(f"**You:** {user}")
+    st.markdown(f"**Assistant:** {bot}")
+    st.divider()