message

app.py

@@ -1,10 +1,11 @@
 import streamlit as st
 from tempfile import NamedTemporaryFile
+import validators
 
 radioButtonList = ["E-commerce CSV (https://www.kaggle.com/datasets/mervemenekse/ecommerce-dataset)",
 "Upload my own CSV",
 "Upload my own PDF",
-"URL Chat with Google Alphabet's 2022 Q2 Earnings Report (https://shorturl.at/csCK3)",
+"URL Chat with Google Alphabet's 2022 Q2 Earnings Report (https://tinyurl.com/f85wujsj)",
 "Enter my own URL"]
 genre = st.radio(
     "Choose dataset to finetune", radioButtonList
@@ -26,15 +27,17 @@ elif genre==radioButtonList[4]:
     pdfCSVURLText = "URL"
 
 isCustomURL = genre==radioButtonList[4]
-title = st.text_input('Enter your own URL', 'https://shorturl.at/csCK3', disabled=not isCustomURL)
-st.write('The current movie title is', title)
+urlInput = st.text_input('Enter your own URL', '', placeholder="Type your URL here (e.g. https://tinyurl.com/f85wujsj)", disabled=not isCustomURL)
+isValidURL = validators.url(urlInput)
 
 isCustomPDF = genre==radioButtonList[1] or genre==radioButtonList[2]
 uploaded_file = st.file_uploader(f"Upload your own {pdfCSVURLText} here", type=pdfCSVURLText.lower(), disabled=not isCustomPDF)
+uploadedFilename = ""
 if uploaded_file is not None:
     with NamedTemporaryFile(dir='.', suffix=f'.{pdfCSVURLText.lower()}') as f:
         f.write(uploaded_file.getbuffer())
-        st.write('The current uploaded file is', f.name)
+        uploadedFilename = f.name
+        # st.write('The current uploaded file is', f.name)
 
 
     # # To read file as bytes:
@@ -56,11 +59,17 @@ if uploaded_file is not None:
 if genre==radioButtonList[0]:
     enableChatBox = True
 elif genre==radioButtonList[1]:
+    enableChatBox = uploadedFilename[-4:]==".csv"
+elif genre==radioButtonList[2]:
+    enableChatBox = uploadedFilename[-4:]==".pdf"
+elif genre==radioButtonList[3]:
     enableChatBox = True
+elif genre==radioButtonList[4]:
+    enableChatBox = isValidURL
 
-title = st.text_input(f'Ask me anything about this {pdfCSVURLText}', '', placeholder="Type your question here (e.g. what was the most sold item?)", disabled=not isCustomURL)
+title = st.text_input(f'Ask me anything about this {pdfCSVURLText}', '', placeholder="Type your question here (e.g. what was the most sold item?)", disabled=not enableChatBox)
 chatWithPDFButton = "CLICK HERE TO START CHATTING"
-if st.button(chatWithPDFButton, type="primary"):
+if st.button(chatWithPDFButton, type="primary", disabled=not enableChatBox):
     pass
 else:
     pass