wissamantoun commited on
Commit
2898bfd
1 Parent(s): 90afd57
Files changed (1) hide show
  1. backend/qa.py +45 -43
backend/qa.py CHANGED
@@ -2,46 +2,48 @@ import streamlit as st
2
 
3
  from qa_utils import annotate_answer, get_qa_answers
4
 
5
- _, col1, _ = st.beta_columns(3)
6
-
7
- with col1:
8
- st.image("is2alni_logo.png", width=200)
9
- st.title("إسألني أي شيء")
10
-
11
- st.markdown(
12
- """
13
- <style>
14
- p, div, input, label {
15
- text-align: right;
16
- }
17
- </style>
18
- """,
19
- unsafe_allow_html=True,
20
- )
21
-
22
- st.sidebar.header("Info")
23
- st.sidebar.image("AraELECTRA.png", width=150)
24
- st.sidebar.write("Powered by [AraELECTRA](https://github.com/aub-mind/arabert)")
25
-
26
- st.sidebar.write("\n")
27
- n_answers = st.sidebar.slider(
28
- "Max. number of answers", min_value=1, max_value=10, value=2, step=1
29
- )
30
-
31
- question = st.text_input("", value="من هو جو بايدن؟")
32
- if "؟" not in question:
33
- question += "؟"
34
-
35
- run_query = st.button("أجب")
36
- if run_query:
37
- # https://discuss.streamlit.io/t/showing-a-gif-while-st-spinner-runs/5084
38
- with st.spinner("... جاري البحث "):
39
- results_dict = get_qa_answers(question)
40
-
41
- if len(results_dict) > 0:
42
- st.write("## :الأجابات هي")
43
- for result in results_dict["results"][:n_answers]:
44
- annotate_answer(result)
45
- f"[**المصدر**](<{result['link']}>)"
46
- else:
47
- st.write("## 😞 ليس لدي جواب")
 
 
 
2
 
3
  from qa_utils import annotate_answer, get_qa_answers
4
 
5
+
6
+ def write():
7
+ _, col1, _ = st.beta_columns(3)
8
+
9
+ with col1:
10
+ st.image("is2alni_logo.png", width=200)
11
+ st.title("إسألني أي شيء")
12
+
13
+ st.markdown(
14
+ """
15
+ <style>
16
+ p, div, input, label {
17
+ text-align: right;
18
+ }
19
+ </style>
20
+ """,
21
+ unsafe_allow_html=True,
22
+ )
23
+
24
+ st.sidebar.header("Info")
25
+ st.sidebar.image("AraELECTRA.png", width=150)
26
+ st.sidebar.write("Powered by [AraELECTRA](https://github.com/aub-mind/arabert)")
27
+
28
+ st.sidebar.write("\n")
29
+ n_answers = st.sidebar.slider(
30
+ "Max. number of answers", min_value=1, max_value=10, value=2, step=1
31
+ )
32
+
33
+ question = st.text_input("", value="من هو جو بايدن؟")
34
+ if "؟" not in question:
35
+ question += "؟"
36
+
37
+ run_query = st.button("أجب")
38
+ if run_query:
39
+ # https://discuss.streamlit.io/t/showing-a-gif-while-st-spinner-runs/5084
40
+ with st.spinner("... جاري البحث "):
41
+ results_dict = get_qa_answers(question)
42
+
43
+ if len(results_dict) > 0:
44
+ st.write("## :الأجابات هي")
45
+ for result in results_dict["results"][:n_answers]:
46
+ annotate_answer(result)
47
+ f"[**المصدر**](<{result['link']}>)"
48
+ else:
49
+ st.write("## 😞 ليس لدي جواب")