test refactor

app.py

@@ -123,37 +123,37 @@ def retriever(query: str, selected_retriever: str):
             placeholder.markdown(text + "▌")
         placeholder.markdown(text)
 
-        if retriever_endpoint == "None":
-            return ""
+    if retriever_endpoint == "None":
+        return ""
 
-        headers = {
-            'Content-Type': 'application/json',
-        }
+    headers = {
+        'Content-Type': 'application/json',
+    }
 
-        json_data = {
-            'claim': query,
-        }
+    json_data = {
+        'claim': query,
+    }
 
-        url = "http://18.227.0.215"
-        port = "8000"
-        response = requests.post(f'{url}:{port}/{retriever_endpoint}', headers=headers, json=json_data)
-        documents = response.json()["Documents"]
+    url = "http://18.227.0.215"
+    port = "8000"
+    response = requests.post(f'{url}:{port}/{retriever_endpoint}', headers=headers, json=json_data)
+    documents = response.json()["Documents"]
 
-        k = 3
-        topk_documents = documents[:k]
+    k = 3
+    topk_documents = documents[:k]
 
-        corpus = '\n\n'.join(topk_documents)
+    corpus = '\n\n'.join(topk_documents)
 
-        retrieved_document_titles = ""
-        for document in topk_documents:
-            try:
-                idx = document.lower().index("abstract:")
-                title = document[:idx].replace('\n', ' ')
-                retrieved_document_titles = retrieved_document_titles + title + ' ||| '
-            except:
-                retrieved_document_titles = retrieved_document_titles + document + ' ||| '
+    retrieved_document_titles = ""
+    for document in topk_documents:
+        try:
+            idx = document.lower().index("abstract:")
+            title = document[:idx].replace('\n', ' ')
+            retrieved_document_titles = retrieved_document_titles + title + ' ||| '
+        except:
+            retrieved_document_titles = retrieved_document_titles + document + ' ||| '
 
-        display_retrieved_documents = f"Retrieved Documents - ||| {retrieved_document_titles}"
+    display_retrieved_documents = f"Retrieved Documents - ||| {retrieved_document_titles}"
 
     with st.chat_message("assistant"):
         placeholder = st.empty()