Update appStore/keyword_search.py

appStore/keyword_search.py

@@ -81,46 +81,44 @@ def app():
         
             file = st.session_state['file']
             if file is not None:
-            with tempfile.NamedTemporaryFile(mode="wb") as temp:
-                bytes_data = file.getvalue()
-                temp.write(bytes_data)
-                file_name = file.name
-                file_path = temp.name
-                
-                st.write("Filename: ", file.name)
+                with tempfile.NamedTemporaryFile(mode="wb") as temp:
+                    bytes_data = file.getvalue()
+                    temp.write(bytes_data)
+                    file_name = file.name
+                    file_path = temp.name
+                    
+                    # load document
+                    documents = pre.load_document(temp.name,file_name)
+                    documents_processed = pre.preprocessing(documents)                                 
+                    pipeline = start_haystack(documents_processed )
                 
-                # load document
-                documents = pre.load_document(temp.name,file_name)
-                documents_processed = pre.preprocessing(documents)                                 
-                pipeline = start_haystack(documents_processed )
-            
 
             
-                with st.spinner("👑 Performing semantic search on"):#+file.name+"..."):
-                    try:
-                        msg = 'Asked ' + question
-                        logging.info(msg)
-                        pipeline = st.session_state['pipeline']
-                        results = ask_question(question,pipeline) 
-                        st.write('## Top Results')
-                        #st.write(results)
-                        for count, result in enumerate(results):
-                            if result["answer"]:
-                                answer, context = result["answer"], result["context"]
-                                start_idx = context.find(answer)
-                                end_idx = start_idx + len(answer)
-                                st.write(
-                                    markdown(context[:start_idx] + str(annotation(body=answer, label="ANSWER", background="#964448", color='#ffffff')) + context[end_idx:]),
-                                    unsafe_allow_html=True,
-                                )
-                                st.markdown(f"**Relevance:** {result['relevance']}")
-                            else:
-                                st.info(
-                                    "🤔    Haystack is unsure whether any of the documents contain an answer to your question. Try to reformulate it!"
-                                )
-                           
-                    except Exception as e:
-                        logging.exception(e)
+                    with st.spinner("👑 Performing semantic search on"):#+file.name+"..."):
+                        try:
+                            msg = 'Asked ' + question
+                            logging.info(msg)
+                            pipeline = st.session_state['pipeline']
+                            results = ask_question(question,pipeline) 
+                            st.write('## Top Results')
+                            #st.write(results)
+                            for count, result in enumerate(results):
+                                if result["answer"]:
+                                    answer, context = result["answer"], result["context"]
+                                    start_idx = context.find(answer)
+                                    end_idx = start_idx + len(answer)
+                                    st.write(
+                                        markdown(context[:start_idx] + str(annotation(body=answer, label="ANSWER", background="#964448", color='#ffffff')) + context[end_idx:]),
+                                        unsafe_allow_html=True,
+                                    )
+                                    st.markdown(f"**Relevance:** {result['relevance']}")
+                                else:
+                                    st.info(
+                                        "🤔    Haystack is unsure whether any of the documents contain an answer to your question. Try to reformulate it!"
+                                    )
+                               
+                        except Exception as e:
+                            logging.exception(e)
                     
             else:
                 st.info("🤔 No document found, please try to upload it at the sidebar!")