Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from ocr_cpu import extract_text_got, clean_text  # Import OCR and text cleaning functions
+from ocr_cpu import extract_text_got, clean_text_with_ai  # Import OCR and AI cleaning functions
 import json
 
 # --- UI Styling ---
@@ -46,8 +46,8 @@ if uploaded_file is not None:
             if not extracted_text.strip():
                 st.warning("No text extracted from the image.")
             else:
-                # Clean the extracted text to remove extra spaces
-                cleaned_text = clean_text(extracted_text)
+                # Clean the extracted text using AI
+                cleaned_text = clean_text_with_ai(extracted_text)
                 st.success("Text extraction and cleaning successful.")
         except Exception as e:
             st.error(f"Error during text extraction: {str(e)}")