Update app.py

app.py

@@ -1,38 +1,63 @@
-import gradio as gr
+import streamlit as st
 import subprocess
 
-examples = [["A qualified alien entered the United States on August 15, 1996. Tell me about their eligibility for Supplemental Security Income (SSI)."],
-            ["""Question: Which of the following statements most accurately reflects the eligibility requirements for Supplemental Security Income (SSI) as outlined in the Alien Eligibility Under Welfare Reform?
-A) To be eligible for SSI, an individual must be a U.S. citizen, with no exceptions for non-citizens.
-B) SSI eligibility is extended to all individuals residing in the United States, regardless of their citizenship or immigration status.
-C) To be eligible for SSI, an individual must be either a U.S. citizen or a qualified alien.
-D) SSI eligibility is determined solely based on financial need, without consideration of citizenship or immigration status."""],
-           ["""Explanate the Correct Answer: C
-Question: Which of the following statements most accurately reflects the eligibility requirements for Supplemental Security Income (SSI) as outlined in the Alien Eligibility Under Welfare Reform?
-A) To be eligible for SSI, an individual must be a U.S. citizen, with no exceptions for non-citizens.
-B) SSI eligibility is extended to all individuals residing in the United States, regardless of their citizenship or immigration status.
-C) To be eligible for SSI, an individual must be either a U.S. citizen or a qualified alien.
-D) SSI eligibility is determined solely based on financial need, without consideration of citizenship or immigration status."""]]
-
-def respond(message, *args, **kwargs):
+st.title("POMS-QA with Knowledge Graph enhancement")
+
+def check_password():
+    """Returns `True` if the user had the correct password."""
+
+    def password_entered():
+        """Checks whether a password entered by the user is correct."""
+        if hmac.compare_digest(st.session_state["password"], "8888p4ss"):
+            st.session_state["password_correct"] = True
+            del st.session_state["password"]  # Don't store the password.
+        else:
+            st.session_state["password_correct"] = False
+
+    # Return True if the password is validated.
+    if st.session_state.get("password_correct", False):
+        return True
+
+    # Show input for password.
+    st.text_input(
+        "Password", type="password", on_change=password_entered, key="password"
+    )
+    if "password_correct" in st.session_state:
+        st.error("😕 Password incorrect")
+    return False
+
+
+if not check_password():
+    st.stop()  # Do not continue if check_password is not True.
+
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+# React to user input
+if prompt := st.chat_input("What is up?"):
+    # Display user message in chat message container
+    with st.chat_message("user"):
+        st.markdown(prompt)
+    # Add user message to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
     command = [
         "python", "-m", "graphrag.query",
         "--root", "./ragtest",
         "--method", "global",
-        message
+        prompt
     ]
-    
-    try:
-        result = subprocess.run(command, capture_output=True, text=True, check=True)
-        return result.stdout
-    except subprocess.CalledProcessError as e:
-        return f"Error: {e.stderr}"
-
-
-demo = gr.ChatInterface(
-    respond,
-    examples=examples)
-
+    result = subprocess.run(command, capture_output=True, text=True, check=True)
+    response = result.stdout
 
-if __name__ == "__main__":
-    demo.launch(auth=("tester", "8888p4ss"))
+    # Display assistant response in chat message container
+    with st.chat_message("assistant"):
+        st.markdown(response)
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": response})