Update app.py

app.py

@@ -10,11 +10,11 @@ import OCR
 import os
 from  OCR import *
 
-# Initialize chat history
+
 if "messages" not in st.session_state:
     st.session_state.messages = [{"role": "Invoice Reader", "content": "Submit an invoice and I will read it."}]
 
-# Display chat messages from history on app rerun
+
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
@@ -22,10 +22,9 @@ HF_TOKEN = os.environ.get("HF_TOKEN")
 USER = "user"
 ASSISTANT = "Invoice Reader"
 
-# Accept file uploads
+
 uploaded_file = st.file_uploader("Upload an invoice", type=["pdf", "png", "jpg", "jpeg"])
 if uploaded_file is not None:
-    # Display uploaded file content
     file_content = uploaded_file.getvalue()
     st.session_state.messages.append({"role": USER, "content": f"Uploaded file: {uploaded_file.name}"})
     with st.chat_message(USER):
@@ -33,10 +32,7 @@ if uploaded_file is not None:
 
     # Preprocess and extract text from image or PDF
     try:
-        if uploaded_file.type == "application/pdf":
-            text = extract_text_from_pdf(file_content)
-        else:
-            text = extract_text_from_image(file_content)
+        text = extract_text_from_image(file_content)
 
         # Extract specific details
         details = extract_invoice_details(text)