Update app.py

app.py

@@ -100,11 +100,29 @@ elif selection=="Adobe Firefly":
     with open('wiki_1.json', 'r') as fr:
         list_1 = json.load(fr)
     
-    with open('wiki_2.json', 'r') as fr:
-        list_2 = json.load(fr)
-    document_name = "Adobe Firefly"
-    section_names = ["Introduction"]*7+["History"]*2
-    ref_doc_indices = np.arange(1,8).tolist() + np.arange(1,3).tolist()
+    # with open('wiki_2.json', 'r') as fr:
+    #     list_2 = json.load(fr)
+
+    tmp_ref_abstract = {}
+
+
+    file_count=0
+    for file in uploaded_files:
+        tmp_filename = file.name
+        tmp_content = file.read().decode('utf-8').strip()
+        tmp_ref_abstract[tmp_filename] = tmp_content
+        file_count+=1
+        
+    document_name = presentation_title
+    section_names = [slide_title]*file_count
+    ref_doc_indices = np.arange(1,file_count+1).tolist()
+
+    list_2 = [
+        {
+            "abstract": "Write the '{}' section of the article titled '{}'.".format(slide_title, presentation_title),
+            "ref_abstract": tmp_ref_abstract
+        }
+    ]
 else:
     with open('wiki_2.json', 'r') as fr:
         list_1 = json.load(fr)
@@ -171,7 +189,7 @@ if st.button('Submit'):
                 st.session_state.summarize_clicked = False                    
             st.session_state.organize_clicked = True
             st.header("Organization of the documents in the narrative")
-            topics_list = ["Introduction", "History", "Document Cloud"]
+            topics_list = [slide_title]
             
             organize_list = []
             ui_organize_list = []
@@ -190,7 +208,7 @@ if st.button('Submit'):
                 #     st.session_state.narrative_clicked = False                                    
                 st.session_state.summarize_clicked = True
                 st.header("Intent-based multi-document summary")
-                topics_list = ["Introduction", "History", "Document Cloud"]
+                topics_list = [slide_title]
                 generate_list = []
                 ui_generate_list = []
                 slides_list = []