Merge pull request #7 from umbc-scify/version_0.1_setup

app.py

@@ -2,9 +2,9 @@ import streamlit as st
 import random
 import time
 
-st.header(" VERIFYING SCIENTIFIC CLAIMS ")
+st.header(" Scientific Claim Verification ")
 
-st.caption("Version 0.1")
+st.caption("Team UMBC-SBU-UT")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -20,7 +20,7 @@ def retriever(query: str):
     with st.chat_message("assistant"):
         placeholder = st.empty()
         text=""
-        message = "Retriving the documents related to the claim..."
+        message = "Retrieving the documents related to the claim..."
         for chunk in message.split():
             text += chunk + " "
             time.sleep(0.05)
@@ -30,7 +30,7 @@ def retriever(query: str):
     # You could return retrieved info here.
     return message
 
-def reasoner(info: str):
+def reasoner(info: list[str]):
     """Simulate a 'reasoner' step, thinking about how to answer."""
     with st.chat_message("assistant"):
         placeholder = st.empty()
@@ -46,7 +46,7 @@ def reasoner(info: str):
     return message
 
 # Accept user input
-if prompt := st.chat_input("40mg/day dosage of folic acid and 2mg/day dosage of vitamin B12 does not affect chronic kidney disease (CKD) progression."):
+if prompt := st.chat_input("Type here"):
     # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
     # Display user message in chat message container
@@ -54,10 +54,10 @@ if prompt := st.chat_input("40mg/day dosage of folic acid and 2mg/day dosage of
         st.markdown(prompt)
 
     #Calling retriever
-    retriever(prompt)
+    retrieved_documents=retriever(prompt)
 
     #Calling reasoner
-    reasoner(prompt)
+    reasoning = reasoner(retrieved_documents)
 
     # Display assistant response in chat message container
     with st.chat_message("assistant"):