Update app.py

app.py

@@ -3,6 +3,8 @@ import openai
 import os
 import base64
 import glob
+import json
+from bs4 import BeautifulSoup
 from datetime import datetime
 from dotenv import load_dotenv
 from openai import ChatCompletion
@@ -22,7 +24,6 @@ def chat_with_model(prompts):
 
 def generate_filename(prompt):
     safe_date_time = datetime.now().strftime("%m_%d_%H_%M")
-
     safe_prompt = "".join(x for x in prompt if x.isalnum())[:50]
     return f"{safe_date_time}_{safe_prompt}.htm"
 
@@ -33,20 +34,40 @@ def create_file(filename, prompt, response):
 def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
         data = file.read()
-    b64 = base64.b64encode(data.encode()).decode()  
+    b64 = base64.b64encode(data.encode()).decode()
     href = f'<a href="data:file/htm;base64,{b64}" target="_blank" download="{os.path.basename(file_path)}">{os.path.basename(file_path)}</a>'
     return href
 
+def read_file_content(file):
+    if file.type == "application/json":
+        content = json.load(file)
+        return str(content)
+    elif file.type == "text/html":
+        content = BeautifulSoup(file, "html.parser")
+        return content.text
+    elif file.type == "application/xml" or file.type == "text/xml":
+        content = BeautifulSoup(file, "lxml")
+        return content.text
+    elif file.type == "text/plain":
+        return file.getvalue().decode()
+    else:
+        return ""
+
 def main():
     st.title("Chat with AI")
 
     prompts = ['']
 
     user_prompt = st.text_area("Your question:", '', height=120)
+    uploaded_file = st.file_uploader("Choose a file", type=["xml", "json", "htm", "txt"])
 
     if user_prompt:
         prompts.append(user_prompt)
 
+    if uploaded_file is not None:
+        file_content = read_file_content(uploaded_file)
+        prompts.append(file_content)
+
     if st.button('Chat'):
         st.write('Chatting with GPT-3...')
         response = chat_with_model(prompts)