Update app.py

app.py

@@ -26,18 +26,26 @@ st.write(f"Your selected model: {model}")
 pipe = pipeline("question-answering", model=model)
 
 # 设置默认的问题和上下文
-default_question = "What is the value of FF?"
+default_property = "FF"
 default_context = "The referential DSSC with Pt CE was also measured under the same conditions, which yields η of 6.66% (Voc= 0.78 V, Jsc= 13.0 mA cm−2, FF = 65.9%)."
 
 # 获取用户输入的问题和上下文
-question = st.text_input("Enter your question here", value=default_question)
-context = st.text_area("Enter the context here", value=default_context, height=300)
+property = st.text_input("Enter your the name of the property: ", value=default_property)
+context = st.text_area("Enter the context: ", value=default_context, height=300)
 
 # 添加一个按钮，用户点击后执行问答
-if st.button('Answer the Question'):
-    if context and question:
+if st.button('Extract the property'):
+    question_1 = f"What is the value of {property}?"
+    if context and question_1:
         out = pipe({
-            'question': question,
+            'question': question_1,
+            'context': context
+        })
+        st.json(out)
+        value = out["answer"]
+        question_2 = f"What material has {property} of {value}?"
+        out = pipe({
+            'question': question_2,
             'context': context
         })
         st.json(out)