Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,17 +9,16 @@ import json
 from langdetect import detect, DetectorFactory
 from gtts import gTTS
 
-# Find the directory this script is in
+# Find this script's folder and the font there
 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 login
 mega = Mega()
 MEGA_EMAIL = os.getenv("MEGA_EMAIL")
 MEGA_PASSWORD = os.getenv("MEGA_PASSWORD")
-
 if not MEGA_EMAIL or not MEGA_PASSWORD:
-    st.error("MEGA credentials not found in environment variables. Please set MEGA_EMAIL and MEGA_PASSWORD.")
+    st.error("MEGA credentials missing. Set MEGA_EMAIL and MEGA_PASSWORD env vars.")
     st.stop()
 
 m = mega.login(MEGA_EMAIL, MEGA_PASSWORD)
@@ -38,7 +37,6 @@ def get_font_for_language(lang_code, size=28):
 
 st.set_page_config(page_title="HeritageVerse", layout="centered")
 st.title("HeritageVerse: Share Your Cultural Stories")
-
 st.write("""
 Welcome to HeritageVerse! Share your stories, proverbs, or memories in any language. We'll detect the language, generate audio, and create a beautiful story card for you.
 """)
@@ -50,7 +48,6 @@ with st.form("story_form"):
 
 if submitted and story.strip():
     st.info("Processing your submission...")
-    # Language detection
     DetectorFactory.seed = 0
     try:
         detected_lang = detect(story)
@@ -64,19 +61,13 @@ if submitted and story.strip():
         st.error(f"Language detection failed: {e}")
         detected_lang = None
 
-    # Use /tmp directory for saving temp files to avoid permission errors
-    tmp_dir = "/tmp"
-    if not os.path.exists(tmp_dir):
-        tmp_dir = "."
-
-    # Generate unique filenames and paths
+    tmp_dir = "/tmp" if os.path.exists("/tmp") else "."
     audio_filename = f"audio_{uuid.uuid4().hex}.mp3"
     audio_path = os.path.join(tmp_dir, audio_filename)
-
     card_filename = f"card_{uuid.uuid4().hex}.png"
     card_path = os.path.join(tmp_dir, card_filename)
 
-    # Generate TTS audio
+    # Text-to-speech
     try:
         tts = gTTS(text=story, lang=detected_lang if detected_lang else 'en')
         tts.save(audio_path)
@@ -88,41 +79,33 @@ if submitted and story.strip():
         st.error(f"TTS generation failed: {e}")
         st.warning("This language might not be supported for audio yet.")
 
-    # Generate image card
+    # Story card image generation
     try:
         title_font = get_font_for_language(detected_lang if detected_lang else 'en', 36)
         body_font = get_font_for_language(detected_lang if detected_lang else 'en', 28)
-
         margin, offset = 40, 80
         max_width = 40
         lines = textwrap.wrap(story, width=max_width)
-
         line_heights = []
         for line in lines:
             bbox = body_font.getbbox(line)
             line_height = bbox[3] - bbox[1]
             line_heights.append(line_height + 8)
-
         total_text_height = sum(line_heights)
         base_height = 400
         extra_height = max(0, total_text_height - (base_height - offset - 60))
         img_height = base_height + extra_height
-
         img = Image.new('RGB', (800, img_height), color=(245, 235, 220))
         draw = ImageDraw.Draw(img)
-
         title_text = "Story Card"
         title_width = draw.textlength(title_text, font=title_font)
         draw.text(((800 - title_width) // 2, 20), title_text, font=title_font, fill=(90, 60, 30))
-
         y = offset
         for i, line in enumerate(lines):
             draw.text((margin, y), line, font=body_font, fill=(60, 40, 20))
             y += line_heights[i]
-
         if name.strip():
             draw.text((margin, y + 20), f"- {name.strip()}", font=body_font, fill=(100, 80, 60))
-
         draw.rectangle([(5, 5), (795, img_height - 5)], outline=(150, 120, 100), width=3)
         img.save(card_path)
         with open(card_path, "rb") as f:
@@ -132,7 +115,7 @@ if submitted and story.strip():
     except Exception as e:
         st.error(f"Story card generation failed: {e}")
 
-    # Upload files to MEGA
+    # Upload to MEGA
     try:
         audio_url = upload_to_mega(audio_path)
         card_url = upload_to_mega(card_path)
@@ -146,7 +129,7 @@ if submitted and story.strip():
         audio_url = None
         card_url = None
 
-    # Manage metadata JSON on MEGA
+    # Metadata
     metadata_file = "stories.json"
     local_metadata_path = os.path.join(tmp_dir, metadata_file)
     try:
@@ -159,7 +142,6 @@ if submitted and story.strip():
             stories = []
     except Exception:
         stories = []
-
     try:
         data_entry = {
             "name": name.strip(),
@@ -170,22 +152,16 @@ if submitted and story.strip():
             "card_url": card_url
         }
         stories.append(data_entry)
-
         with open(local_metadata_path, "w", encoding="utf-8") as f:
             json.dump(stories, f, ensure_ascii=False, indent=2)
-
         m.upload(local_metadata_path)
         st.success("Metadata updated on MEGA!")
     except Exception as e:
         st.error(f"Failed to update metadata JSON on MEGA: {e}")
 
-    # Clean up local temp files
+    # Cleanup
     try:
-        if os.path.exists(audio_path):
-            os.remove(audio_path)
-        if os.path.exists(card_path):
-            os.remove(card_path)
-        if os.path.exists(local_metadata_path):
-            os.remove(local_metadata_path)
-    except:
-        pass
+        if os.path.exists(audio_path): os.remove(audio_path)
+        if os.path.exists(card_path): os.remove(card_path)
+        if os.path.exists(local_metadata_path): os.remove(local_metadata_path)
+    except: pass