Update app.py

app.py

@@ -1,12 +1,51 @@
+# import streamlit as st
+# import easyocr
+# from PIL import Image
+# import numpy as np
+
+# # Initialize EasyOCR Reader
+# reader = easyocr.Reader(['en'])  # You can specify the language(s) here
+
+# st.title("Image to Text Extraction Using EasyOCR")
+
+# # File uploader for images
+# uploaded_file = st.file_uploader("Choose an image... (JPEG, PNG)", type=["jpg", "jpeg", "png"])
+
+# if uploaded_file is not None:
+#     # Load the image using PIL
+#     image = Image.open(uploaded_file)
+#     st.image(image, caption='Uploaded Image', use_container_width=True)
+#     st.write("")
+
+#     # Convert image to numpy array
+#     image_np = np.array(image)
+
+#     # Use EasyOCR to do OCR on the image
+#     results = reader.readtext(image_np)
+
+#     # Extract and display the detected text
+#     st.subheader("Extracted Text:")
+
+#     if results:
+#         for (bbox, text, prob) in results:
+#             st.write(f"Detected Text: '{text}' with confidence {prob:.2f}")
+#     else:
+#         st.write("No text detected.")
+
+
+
+
+
 import streamlit as st
 import easyocr
 from PIL import Image
 import numpy as np
+import requests
 
 # Initialize EasyOCR Reader
-reader = easyocr.Reader(['en'])  # You can specify the language(s) here
+reader = easyocr.Reader(['en'])
 
-st.title("Image to Text Extraction Using EasyOCR")
+st.title("Image to Text Extraction and Translation Using EasyOCR and Translation API")
 
 # File uploader for images
 uploaded_file = st.file_uploader("Choose an image... (JPEG, PNG)", type=["jpg", "jpeg", "png"])
@@ -24,10 +63,32 @@ if uploaded_file is not None:
     results = reader.readtext(image_np)
 
     # Extract and display the detected text
-    st.subheader("Extracted Text:")
-
+    st.subheader("Extracted English Text:")
+    extracted_texts = []
     if results:
         for (bbox, text, prob) in results:
             st.write(f"Detected Text: '{text}' with confidence {prob:.2f}")
+            extracted_texts.append(text)
     else:
         st.write("No text detected.")
+
+    # Combine all extracted texts into a single string
+    english_text = ' '.join(extracted_texts)
+
+    # Translate English text to Persian using an API
+    st.subheader("Translated Persian Text:")
+    if english_text:
+        with st.spinner("Translating text..."):
+            # Make a request to the translation API
+            translation_response = requests.post('https://translate.sravn.io/translate', data={
+                'q': english_text,
+                'source': 'en',
+                'target': 'fa'
+            })
+            if translation_response.status_code == 200:
+                persian_text = translation_response.json()['translatedText']
+                st.write(persian_text)
+            else:
+                st.write("Translation failed. Please try again later.")
+    else:
+        st.write("No text to translate.")