Upload 4 files

README.md

@@ -1,9 +1,13 @@
 ---
-title: Unlimited Research Group of AI
-emoji: 🚀
-colorFrom: indigo
+title: Ura Introduciton
+emoji: 📉
+colorFrom: blue
 colorTo: blue
 sdk: streamlit
-sdk_version: 1.27.0
+sdk_version: 1.27.2
+app_file: app.py
 pinned: false
----
+license: mit
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,72 @@
+import streamlit as st
+import requests
+import json
+import os
+
+url = os.environ["SERVER_API"]
+headers = {
+    'Content-Type': 'application/json'
+}
+
+if 'context' not in st.session_state:
+    st.session_state.context = ""
+
+
+def answer_query(query):
+    payload = json.dumps({
+        "context": f"{st.session_state.context}",
+        "question": f"{query}",
+        "lang": "vi"
+    })
+    response = requests.request(
+        "POST",
+        url,
+        headers=headers,
+        data=payload,
+        timeout=300
+    )
+    answer = response.json()["answer"]
+    return answer
+
+
+def main():
+    st.set_page_config(layout="wide")
+    st.image(["Logo BK.png", "Logo CSE.png"], width=120)
+    st.title("URA-LLaMa Playground")
+
+    st.subheader("Contexts")
+    use_manual_context = st.checkbox('Use manual context', value=True)
+    if use_manual_context:
+        context = st.text_area(
+            "Enter context", 
+            value="Chương trình Khoa học máy tính của Khoa Khoa học và Kỹ thuật Máy tính (CSE), trường Đại học Bách Khoa (HCMUT) - Đại học Quốc gia thành phố Hồ Chí Minh được kiểm định bởi ABET.",
+            height=200,
+            max_chars=5000
+        )
+        if context != st.session_state.context:
+            st.session_state.context = ""
+
+            if context != "":
+                with st.spinner('Processing contexts...'):
+                    st.session_state.context = context
+    else:
+        st.session_state.context = ""
+
+    st.subheader("Question")
+    question = st.text_input("Enter your question", placeholder="ABET là gì?")
+    button = st.button("Submit")
+
+    # Display the answer and additional context
+    st.subheader('Answer')
+    if button and question != "":
+        with st.spinner('Finding answers...'):
+            answer = answer_query(question)
+    else:
+        answer = ""
+    st.write(answer)
+
+    st.write("Powered by URA-LLaMa 7B")
+
+
+if __name__ == "__main__":
+    main()