Update app.py

app.py

@@ -13,114 +13,149 @@ from reportlab.lib.utils import ImageReader
 import mistune
 from gtts import gTTS
 
-# Page setup
+# Page config
 st.set_page_config(page_title="PDF & Code Interpreter", layout="wide", page_icon="🚀")
 
-# Tabs
-tab1, tab2 = st.tabs(["📄 PDF Composer","🧪 CodeInterpreter"])
+def delete_asset(path):
+    try:
+        os.remove(path)
+    except:
+        pass
+    st.rerun()
+
+# Tabs setup
+tab1, tab2 = st.tabs(["📄 PDF Composer", "🧪 CodeInterpreter"])
 
-# --- Tab 1: PDF Composer ---
 with tab1:
-    st.title("🖼️📝 PDF Composer & Voice Generator")
-    # Upload Markdown
-    md_file = st.file_uploader("Upload Markdown (.md)", type=["md"] )
-    md_text = ""
+    st.header("📄 PDF Composer & Voice Generator 🚀")
+
+    md_file = st.file_uploader("Upload Markdown (.md)", type=["md"])
     if md_file:
         md_text = md_file.getvalue().decode("utf-8")
     else:
-        md_text = st.text_area("Or enter markdown text", height=200)
-    # Convert markdown to plain text
-    plain = mistune.HTMLRenderer().render(mistune.create_markdown()) if not md_text.strip() else mistune.markdown(md_text)
-
-    # Generate audio
-    if st.button("🔊 Generate Voice Track MP3"):
-        tts = gTTS(text=re.sub(r'<[^>]+>','', plain), lang='en')
-        fname = f"voice_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp3"
-        tts.save(fname)
-        st.audio(fname)
-
-    # Upload Images
-    imgs = st.file_uploader("Upload Images", type=["png","jpg","jpeg"], accept_multiple_files=True)
-    # Reorder images
+        md_text = st.text_area("Or enter markdown text directly", height=200)
+
+    renderer = mistune.HTMLRenderer()
+    markdown = mistune.create_markdown(renderer=renderer)
+    html = markdown(md_text or "")
+    plain_text = re.sub(r'<[^>]+>', '', html)
+
+    if st.button("🔊 Generate Voice MP3 from Text"):
+        if plain_text.strip():
+            voice_file = f"voice_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp3"
+            tts = gTTS(text=plain_text, lang='en')
+            tts.save(voice_file)
+            st.audio(voice_file)
+        else:
+            st.warning("No text to generate voice from.")
+
+    imgs = st.file_uploader("Upload Images for PDF", type=["png", "jpg", "jpeg"], accept_multiple_files=True)
+    ordered_images = []
     if imgs:
-        df = pd.DataFrame([{"name":f.name,"order":i} for i,f in enumerate(imgs)])
-        edited = st.experimental_data_editor(df, num_rows="fixed")
-        ordered = [imgs[df.loc[i,'name'] == f.name][0] for i in range(len(df))]
-    else:
-        ordered = []
+        df_imgs = pd.DataFrame([{"name": f.name, "order": i} for i, f in enumerate(imgs)])
+        edited = st.data_editor(df_imgs, use_container_width=True)
+        for _, row in edited.sort_values('order').iterrows():
+            for f in imgs:
+                if f.name == row['name']:
+                    ordered_images.append(f)
+                    break
 
-    # Generate PDF
-    if st.button("🖋️ Generate PDF"):
+    if st.button("🖋️ Generate PDF with Markdown & Images"):
         buf = io.BytesIO()
         c = canvas.Canvas(buf)
-        # Add markdown text
         y = 800
-        for line in re.sub(r'<[^>]+>','', plain).split('\n'):
-            c.drawString(40,y,line)
+        for line in plain_text.split('\n'):
+            c.drawString(40, y, line)
             y -= 15
-            if y<50:
-                c.showPage(); y=800
-        # Add images
-        for imgf in ordered:
-            img = Image.open(imgf)
-            w,h = img.size
-            c.showPage()
-            c.drawImage(ImageReader(img),0,0,w,h)
-        c.save(); buf.seek(0)
-        pname = f"document_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
-        st.download_button("⬇️ Download PDF", data=buf, file_name=pname, mime="application/pdf")
-
-    # Show existing PDF files
-    st.markdown("### Available PDFs")
-    pdfs = sorted(glob.glob("*.pdf"))
-    for p in pdfs:
-        st.write(p, st.button("Download", key=p, args=(), kwargs={}, help="Download PDF"))
-
-# --- Tab 2: CodeInterpreter ---
+            if y < 50:
+                c.showPage()
+                y = 800
+        for img_f in ordered_images:
+            try:
+                img = Image.open(img_f)
+                w, h = img.size
+                c.showPage()
+                c.drawImage(ImageReader(img), 0, 0, w, h, preserveAspectRatio=True, mask='auto')
+            except:
+                continue
+        c.save()
+        buf.seek(0)
+        pdf_name = f"doc_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
+        st.download_button("⬇️ Download PDF", data=buf, file_name=pdf_name, mime="application/pdf")
+
+    st.markdown("---")
+    st.subheader("📂 Available Assets")
+    assets = sorted(glob.glob("*.*"))
+    for a in assets:
+        ext = a.split('.')[-1].lower()
+        cols = st.columns([3, 1, 1])
+        cols[0].write(a)
+        if ext == 'pdf':
+            with open(a, 'rb') as fp:
+                cols[1].download_button("📥", data=fp, file_name=a, mime="application/pdf")
+        elif ext == 'mp3':
+            cols[1].audio(a)
+        cols[2].button("🗑️", key=f"del_{a}", on_click=delete_asset, args=(a,))
+
 with tab2:
+    st.header("🧪 Python Code Executor & Demo")
     import io, sys
     from contextlib import redirect_stdout
+
     DEFAULT_CODE = '''import streamlit as st
 import random
 
-st.title("🧪 Test Program")
-st.markdown("Welcome to this simple test application!")
+st.title("📊 Demo App")
+st.markdown("Random number and color demo")
 
 col1, col2 = st.columns(2)
 with col1:
-    number = st.number_input("Enter a number:",1,100,50)
-    mult = st.slider("Select multiplier:",1,10,5)
-    if st.button("Calculate"):
-        st.success(f"Result: {number*mult}")
+    num = st.number_input("Number:", 1, 100, 10)
+    mul = st.slider("Multiplier:", 1, 10, 2)
+    if st.button("Calc"):
+        st.write(num * mul)
 with col2:
-    color = st.color_picker("Pick a color","#00f900")
-    st.markdown(f'<div style="background-color:{color};padding:20px;">Color {color}</div>',unsafe_allow_html=True)
-    if st.button("Random Number"):
-        st.balloons(); st.write(random.randint(1,100))
-'''
-    def extract_python_code(md): return re.findall(r"```python\s*(.*?)```",md,re.DOTALL)
-    def execute_code(code):
-        buf = io.StringIO(); lv={}
+    color = st.color_picker("Pick color","#ff0000")
+    st.markdown(f'<div style="background:{color};padding:10px;">Color</div>', unsafe_allow_html=True)
+'''  # noqa
+
+    def extract_python_code(md: str) -> list:
+        return re.findall(r"```python\s*(.*?)```", md, re.DOTALL)
+
+    def execute_code(code: str) -> tuple:
+        buf = io.StringIO()
+        local_vars = {}
         try:
-            with redirect_stdout(buf): exec(code,{},lv)
-            return buf.getvalue(),None
-        except Exception as e: return None,str(e)
-    st.title("📝 Python Code Executor")
-    up = st.file_uploader("Upload .py or .md",type=['py','md'])
-    code = DEFAULT_CODE if 'code' not in st.session_state else st.session_state.code
+            with redirect_stdout(buf):
+                exec(code, {}, local_vars)
+            return buf.getvalue(), None
+        except Exception as e:
+            return None, str(e)
+
+    up = st.file_uploader("Upload .py or .md", type=['py', 'md'])
+    if 'code' not in st.session_state:
+        st.session_state.code = DEFAULT_CODE
+
     if up:
-        txt=up.getvalue().decode();
-        if up.type=='text/markdown':
-            blocks=extract_python_code(txt)
-            code=blocks[0] if blocks else ''
-        else: code=txt
-        st.code(code,language='python')
+        text = up.getvalue().decode()
+        if up.type == 'text/markdown':
+            codes = extract_python_code(text)
+            st.session_state.code = codes[0] if codes else ''
+        else:
+            st.session_state.code = text
+        st.code(st.session_state.code, language='python')
     else:
-        code=st.text_area("Code:",value=code,height=300)
-    st.session_state.code=code
-    if st.button("▶️ Run Code"):
-        out,err=execute_code(code)
-        if err: st.error(err)
-        elif out: st.code(out)
-        else: st.success("Executed with no output.")
-    if st.button("🗑️ Clear Code"): st.session_state.code=''; st.rerun()
+        st.session_state.code = st.text_area("💻 Code Editor", value=st.session_state.code, height=300)
+
+    c1, c2 = st.columns([1, 1])
+    if c1.button("▶️ Run Code"):
+        out, err = execute_code(st.session_state.code)
+        if err:
+            st.error(err)
+        elif out:
+            st.code(out)
+        else:
+            st.success("Executed with no output.")
+    if c2.button("🗑️ Clear Code"):
+        st.session_state.code = ''
+        st.rerun()