Update app.py

app.py

@@ -281,18 +281,11 @@ def create_file(filename, prompt, response, should_save=True):
     if not should_save:
         return
     base_filename, ext = os.path.splitext(filename)
-    #has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
-    has_python_code = False
+    has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
     if ext in ['.txt', '.htm', '.md']:
         with open(f"{base_filename}.md", 'w') as file:
-            try:
-                content = prompt.strip() + '\r\n' + response
-                file.write(content)
-            except:
-                st.write('Skipping write of file.  Prompt and Response below.')
-                st.write(prompt)
-                st.write(response)
-        
+            content = prompt.strip() + '\r\n' + response
+            file.write(content)
         if has_python_code:
             python_code = re.findall(r"```python([\s\S]*?)```", response)[0].strip()
             with open(f"{base_filename}-Code.py", 'w') as file:
@@ -310,11 +303,8 @@ def divide_document(document, max_length):
 @st.cache_resource
 def get_table_download_link(file_path):
     with open(file_path, 'r') as file:
-        try:
-            data = file.read()
-        except:
-            st.write('')
-            return file_path    
+        data = file.read()
+   
     b64 = base64.b64encode(data.encode()).decode()  
     file_name = os.path.basename(file_path)
     ext = os.path.splitext(file_name)[1]  # get the file extension
@@ -436,16 +426,13 @@ def pdf2txt(docs):
     for file in docs:
         file_extension = extract_file_extension(file)
         st.write(f"File type extension: {file_extension}")
-        try:
-            if file_extension.lower() in ['py', 'txt', 'html', 'htm', 'xml', 'json']:
-                text += file.getvalue().decode('utf-8')
-            elif file_extension.lower() == 'pdf':
-                from PyPDF2 import PdfReader
-                pdf = PdfReader(BytesIO(file.getvalue()))
-                for page in range(len(pdf.pages)):
-                    text += pdf.pages[page].extract_text() # new PyPDF2 syntax
-        except Exception as e:
-            st.write(f"Error processing file {file.name}: {e}")
+        if file_extension.lower() in ['py', 'txt', 'html', 'htm', 'xml', 'json']:
+            text += file.getvalue().decode('utf-8')
+        elif file_extension.lower() == 'pdf':
+            from PyPDF2 import PdfReader
+            pdf = PdfReader(BytesIO(file.getvalue()))
+            for page in range(len(pdf.pages)):
+                text += pdf.pages[page].extract_text() # new PyPDF2 syntax
     return text
 
 def txt2chunks(text):