Update app.py

app.py

@@ -274,6 +274,15 @@ def search_arxiv(query):
     st.write(f"Elapsed time: {elapsed_seconds:.2f} seconds")
     filename = generate_filename(query, "md")
     create_file(filename, query, results, should_save)
+
+    
+    # After search completes and generates results
+    result = "Your search results in Markdown format"  # Replace with actual results
+    
+    # Call the function to redraw sidebar and show results
+    redraw_sidebar_after_search(result)
+
+    
     return results
 
 def download_pdfs_and_generate_html(urls):
@@ -614,6 +623,17 @@ def FileSidebar():
         elif st.session_state.action == 'edit':
             edit_file(st.session_state.current_file)
 
+    # Show the result MD from the search
+    if 'search_result' in st.session_state:
+        st.markdown("## Search Result")
+        st.markdown(st.session_state.search_result)
+
+# Function to be called after search completes
+def redraw_sidebar_after_search(search_result):
+    st.session_state.search_result = search_result
+    st.rerun()
+    
+
 def view_file(file):
     st.header(f"Viewing: {file}")
     file_extension = os.path.splitext(file)[1].lower()
@@ -634,7 +654,12 @@ def edit_file(file):
     file_extension = os.path.splitext(file)[1].lower()
     try:
         if file_extension == '.pdf':
-            st.warning("PDF files cannot be edited directly. Please download and edit locally.")
+            md_content = pdf_to_markdown(file)
+            md_file = file.replace('.pdf', '.md')
+            new_content = st.text_area("Edit PDF content (as Markdown)", md_content, height=400)
+            if st.button("Save as Markdown"):
+                new_file_path = update_file_with_timestamp(md_file, new_content)
+                st.success(f"File saved as Markdown! New file: {new_file_path}")
         elif file_extension == '.html':
             with open(file, 'r', encoding='utf-8') as f:
                 content = f.read()
@@ -651,7 +676,18 @@ def edit_file(file):
                 st.success(f"File updated successfully! New file: {new_file_path}")
     except Exception as e:
         st.error(f"Error editing {file}: {str(e)}")
-        
+
+def pdf_to_markdown(file_path):
+    try:
+        with open(file_path, "rb") as f:
+            pdf_reader = PyPDF2.PdfReader(f)
+            markdown_content = ""
+            for page in pdf_reader.pages:
+                markdown_content += page.extract_text() + "\n\n"
+        return markdown_content
+    except Exception as e:
+        st.error(f"Error converting PDF to Markdown: {str(e)}")
+        return ""        
 def display_pdf(file_path):
     try:
         with open(file_path, "rb") as f: