Changes

app.py

@@ -12,7 +12,8 @@ logging.basicConfig(
     force=True,
 )
 
-p = None
+p_1 = None
+p_2 = None
 
 
 def app_init():
@@ -24,16 +25,36 @@ def app_init():
     ds.update_embeddings(indexing_pipeline.get_node("Retriever"))
     ds.save(config_path="my_faiss_config.json", index_path="my_faiss_index.faiss")
 
-    global p
-    p = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query")
+    global p_1
+    p_1 = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query_1")
+
+    global p_2
+    p_2 = Pipeline.load_from_yaml("pipeline.yaml", pipeline_name="query_2")
 
 
 def main():
     app_init()
     st.title("Haystack Demo")
     input = st.text_input("Query ...")
-    result = p.run(input)
-    st.text()
+
+    query_type = st.radio("Type",
+                          ("Retrieval Augmented", "Retrieval Augmented with Sources",
+                           "Retrieval Augmented with Web Search"))
+
+
+    col_1, col_2 = st.columns(2)
+
+    with col_1:
+        st.text("PLAIN")
+        answers = p_1.run(input)["answers"]
+        for ans in answers:
+            st.text(ans.answer)
+
+    with col_2:
+        st.write(query_type.upper())
+        answers = p_2.run(input)["answers"]
+        for ans in answers:
+            st.text(ans.answer)
 
 
 if __name__ == "__main__":

pipeline.yaml

@@ -20,22 +20,31 @@ components:
       frequency_penalty: 0.1 # Positive values penalize new tokens based on their existing frequency in the text so far.
       presence_penalty: 0.1 #  Positive values penalize new tokens based on whether they have already appeared in the text.
       top_k: 3 # The number of results to return
+      api_key: "loaded-from-env"
 
   - name: TextFileConverter
     type: TextConverter
 pipelines:
-  - name: query
+  - name: query_1
     nodes:
       - inputs: [Query]
         name: Retriever
       - inputs: [Retriever]
         name: PromptNode
+
+  - name: query_2
+    nodes:
+      - inputs: [ Query ]
+        name: Retriever
+      - inputs: [ Retriever ]
+        name: PromptNode
+
   - name: indexing
     nodes:
       - name: TextFileConverter
-        inputs: [File]
+        inputs: [ File ]
       - name: Retriever
-        inputs: [TextFileConverter]
+        inputs: [ TextFileConverter ]
       - name: DocumentStore
-        inputs: [Retriever]
+        inputs: [ Retriever ]
 version: ignore