Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,6 +9,10 @@ import json
 from langdetect import detect, DetectorFactory
 from gtts import gTTS
 
+# Find the directory this script is in
+this_dir = os.path.dirname(os.path.abspath(__file__))
+font_path = os.path.join(this_dir, "NotoSansTelugu-VariableFont_wdth,wght.ttf")
+
 # Initialize Mega client and login securely
 mega = Mega()
 MEGA_EMAIL = os.getenv("MEGA_EMAIL")
@@ -27,7 +31,7 @@ def upload_to_mega(filepath):
 
 def get_font_for_language(lang_code, size=28):
     try:
-        return ImageFont.truetype("NotoSansTelugu-VariableFont_wdth,wght.ttf", size)
+        return ImageFont.truetype(font_path, size)
     except Exception as e:
         st.warning(f"Could not load font: {e}")
         return ImageFont.load_default()
@@ -63,7 +67,6 @@ if submitted and story.strip():
     # Use /tmp directory for saving temp files to avoid permission errors
     tmp_dir = "/tmp"
     if not os.path.exists(tmp_dir):
-        # For local testing where /tmp might not exist, fallback to current directory
         tmp_dir = "."
 
     # Generate unique filenames and paths
@@ -125,7 +128,6 @@ if submitted and story.strip():
         with open(card_path, "rb") as f:
             st.download_button("Download Story Card", f, file_name=card_filename)
         st.image(card_path, caption="Your Story Card", use_container_width=True)
-
         st.success("Story card generated!")
     except Exception as e:
         st.error(f"Story card generation failed: {e}")
@@ -137,7 +139,7 @@ if submitted and story.strip():
         st.success("Files uploaded to MEGA successfully!")
         st.audio(audio_url)
         st.markdown(f"[Download Audio]({audio_url})")
-        st.image(card_url, caption="Your Story Card")
+        st.image(card_url, caption="Your Story Card", use_container_width=True)
         st.markdown(f"[Download Card]({card_url})")
     except Exception as e:
         st.error(f"Failed to upload files to MEGA: {e}")