Implement first POC

app.py

@@ -15,6 +15,7 @@ from utils import blob_service_client, upload_to_azure, download_from_azure, del
 load_dotenv()
 
 # Streamlit UI
+st.set_page_config(layout="wide")
 st.title("Azure Translation Tools")
 uploaded_files = st.file_uploader("Upload files to start the process", accept_multiple_files=True)
 
@@ -109,11 +110,12 @@ if uploaded_files and submit:
 
             file_name = uploaded_file.name
             file_content = uploaded_file.read()
+            file_type = file_name.split('.')[-1]
 
             # Check file extension to determine translation method
-            if uploaded_file.name.split('.')[-1] in ['txt', 'tsv', 'tab', 'csv', 'html', 'htm', 'mthml', 'mht', 'pptx', 'xlsx', 'docx', 'msg', 'xlf', 'xliff']:
+            if file_type in ['txt', 'tsv', 'tab', 'csv', 'html', 'htm', 'mthml', 'mht', 'pptx', 'xlsx', 'docx', 'msg', 'xlf', 'xliff']:
                 result, response = process_sync(file_name, file_content)
-            elif uploaded_file.name.split('.')[-1] in ['pdf', 'odt', 'odp', 'ods', 'rtf']:
+            elif file_type in ['pdf', 'odt', 'odp', 'ods', 'rtf']:
                 result, response = process_async(file_name, file_content)
 
             # Calculate duration
@@ -127,14 +129,28 @@ if uploaded_files and submit:
             else:
                 st.error(f"Failed to translate {uploaded_file.name} with status code {response.status_code}: {response.text} (Time taken: {duration:.2f} seconds)")
 
-            # Display the original and translated files side by side
-            col1, col2 = st.columns(2)
-            with col1:
-                st.write(f"Original File: {uploaded_file.name}")
-                pdf_viewer(file_content)
-            with col2:
-                st.write(f"Translated File: {lang_name}-translated-{uploaded_file.name}")
-                pdf_viewer(response)
+            if file_type == 'pdf':
+                # Display the original and translated files side by side
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.write(f"Original File: {uploaded_file.name}")
+                    pdf_viewer(file_content)
+                with col2:
+                    st.write(f"Translated File: {lang_name}-translated-{uploaded_file.name}")
+                    pdf_viewer(response)
+            elif file_type == 'docx':
+                pass
+            elif file_type == 'txt':
+                # Display the original and translated files side by side
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.write(f"Original File: {uploaded_file.name}")
+                    st.divider()
+                    st.write(file_content)
+                with col2:
+                    st.write(f"Translated File: {lang_name}-translated-{uploaded_file.name}")
+                    st.divider()
+                    st.write(response)
 
             # Update progress bar based on completed translations
             progress = (idx + 1) / len(uploaded_files)