Update app.py

app.py

@@ -36,22 +36,28 @@ if (st.button('بحث', disabled=False)):
     st.dataframe(df)
         
 if (st.button('استعمال الرسم البياني', disabled=False)):
+    
     tmt = {} 
     VocMap = './voc.csv'
     ScoreMap = './BM25.csv'
     
-    df3 = pd.read_csv(VocMap, delimiter='\t')
-    df_g = pd.read_csv(ScoreMap, delimiter='\t')
-    df_g.set_index(['ID1','ID2'], inplace=True)
-    
-    df_in = pd.read_csv(ScoreMap, delimiter='\t')
-    df_in.set_index(['ID1'], inplace=True)
-    
+    @st.cache
+    def reading_df(path1, path2):
+        df_voc = pd.read_csv(path1, delimiter='\t')
+        df_graph = pd.read_csv(path2, delimiter='\t')
+        df_graph.set_index(['ID1','ID2'], inplace=True)
+        df_gr = pd.read_csv(ScoreMap, delimiter='\t')
+        df_gr.set_index(['ID1'], inplace=True)
+        return df_voc, df_graph, df_gr
+        
+    df3, df_g, df_in = reading_df(path1, path2)
+
+    @st.cache
     def Query2id(voc, query):
         return [voc.index[voc['word'] == word].values[0] for word in query.split()]
     
     id_list = Query2id(df3, sent)
-    
+    @st.cache
     def setQueriesVoc(df, id_list):
         res = []
         for e in id_list: