Update app.py

app.py

@@ -8,6 +8,8 @@ import openai
 from sklearn.neighbors import NearestNeighbors
 import os
 import time
+import csv
+from io import StringIO
 
 def download_pdf(url, output_path):
     urllib.request.urlretrieve(url, output_path)
@@ -132,6 +134,9 @@ st.markdown(description)
 
 openAI_key = st.sidebar.text_input('API Key', value='sk-')
 
+data_section = st.sidebar.text_area("Paste Data:")
+paste_data = st.sidebar.button("Paste Data")
+
 add_row = st.sidebar.button("Add row")
 row_count = st.session_state.get("row_count", 1)
 
@@ -139,13 +144,25 @@ if add_row:
     row_count += 1
     st.session_state.row_count = row_count
 
+if paste_data:
+    data = StringIO(data_section.strip())
+    reader = csv.reader(data, delimiter=',', quotechar='"')
+    urls_questions = [row for row in reader]
+
+    row_count = len(urls_questions)
+    st.session_state.row_count = row_count
+
+    for i, url_question in enumerate(urls_questions):
+        st.session_state[f"url{i}"] = url_question[0]
+        st.session_state[f"question{i}"] = url_question[1]
+
 for i in range(row_count):
     col1, col2, col3, col4 = st.columns(4)
 
     with col1:
-        url = st.text_input(f'PDF URL {i+1}', key=f'url{i}')
+        url = st.text_input(f'PDF URL {i+1}', key=f'url{i}', value=st.session_state.get(f'url{i}', ''))
     with col2:
-        question = st.text_input(f'Question {i+1}', key=f'question{i}')
+        question = st.text_input(f'Question {i+1}', key=f'question{i}', value=st.session_state.get(f'question{i}', ''))
     with col3:
         # Initialize session state for answer if not already done
         if f'session_answer{i}' not in st.session_state: