Update app.py

app.py

@@ -7,13 +7,22 @@ from io import BytesIO
 from deep_translator import GoogleTranslator
 import torch
 
-translated_caption = GoogleTranslator(source='auto', target=languages[selected_language]).translate(cap)
-
+# Streamlit page config
 st.set_page_config(page_title="AI Image Caption Generator", page_icon="🖼️")
 
 st.title("🖼️ AI Image Caption Generator")
 st.write("Upload an image or paste a Google Image URL to get multiple captions generated by AI, in your preferred language!")
 
+# Language options
+languages = {
+    "English": "en",
+    "Urdu": "ur",
+    "Hindi": "hi",
+    "French": "fr",
+    "Spanish": "es",
+    "Arabic": "ar"
+}
+
 # Image input options
 upload_option = st.radio("Choose Image Input Method:", ("Upload from Computer", "Paste Image URL"))
 
@@ -25,16 +34,7 @@ if upload_option == "Upload from Computer":
 else:
     image_url = st.text_input("Paste Image URL (must be direct link ending with .jpg/.png/.jpeg)")
 
-# Language options
-languages = {
-    "English": "en",
-    "Urdu": "ur",
-    "Hindi": "hi",
-    "French": "fr",
-    "Spanish": "es",
-    "Arabic": "ar"
-}
-
+# Language selection
 selected_language = st.selectbox("🌐 Choose Output Language", list(languages.keys()))
 
 # Load image based on input
@@ -81,7 +81,7 @@ if image:
 
     for idx, cap in enumerate(captions):
         try:
-            translated_caption = translator.translate(cap, dest=languages[selected_language]).text
+            translated_caption = GoogleTranslator(source='auto', target=languages[selected_language]).translate(cap)
         except Exception as e:
             translated_caption = f"(Translation Error: Showing English) {cap}"