Create app.py

app.py

@@ -0,0 +1,163 @@
+import streamlit as st
+from pptx import Presentation
+from pptx.util import Inches, Pt
+from pptx.dml.color import RGBColor
+import io, tempfile, os, re
+from transformers import pipeline
+
+# ── CONFIG ─────────────────────────────────────────────────────
+st.set_page_config(page_title="PPTX Smart Enhancer", layout="wide")
+MODEL_ID = "Faisalkhany/newfinetune"  # ← your HF model
+pipe = pipeline("text2text-generation", model=MODEL_ID)
+
+TEMPLATES = {
+    "Office Default": {
+        "font": "Calibri",
+        "colors": {"primary":"#2B579A","secondary":"#5B9BD5","text":"#000000","background":"#FFFFFF"},
+        "sizes": {"title":44,"heading":32,"body":18}, "background":True
+    },
+    "Modern Office": {
+        "font": "Segoe UI",
+        "colors": {"primary":"#404040","secondary":"#A5A5A5","text":"#FFFFFF","background":"#121212"},
+        "sizes": {"title":36,"heading":24,"body":16}, "background":True
+    },
+    "Corporate": {
+        "font":"Arial",
+        "colors":{"primary":"#1976D2","secondary":"#BBDEFB","text":"#212121","background":"#FAFAFA"},
+        "sizes":{"title":40,"heading":28,"body":18}, "background":False
+    }
+}
+FONT_CHOICES = {f:f for f in ["Arial","Calibri","Times New Roman","Segoe UI","Verdana","Georgia"]}
+
+# ── HELPERS ────────────────────────────────────────────────────
+def chunk_slide_text(text, max_words=200):
+    words = text.split()
+    return [" ".join(words[i:i+max_words]) for i in range(0, len(words), max_words)]
+
+def get_ai_response(prompt: str) -> str:
+    out = pipe(prompt, max_length=512, truncation=True)
+    return out[0]["generated_text"]
+
+def extract_slide_text(slide) -> str:
+    parts = []
+    for shp in slide.shapes:
+        if hasattr(shp, "text_frame"):
+            for p in shp.text_frame.paragraphs:
+                if p.text.strip():
+                    parts.append(p.text.strip())
+    return "\n".join(parts)
+
+def split_formatted_runs(text):
+    segments, cur, bold, italic = [], [], False, False
+    parts = re.split(r'(\*|_)', text)
+    for tok in parts:
+        if tok in ("*","_"):
+            if cur:
+                segments.append({"text":"".join(cur),"bold":bold,"italic":italic})
+                cur=[]
+            if tok=="*": bold=not bold
+            else: italic=not italic
+        else:
+            cur.append(tok)
+    if cur:
+        segments.append({"text":"".join(cur),"bold":bold,"italic":italic})
+    return segments
+
+def apply_formatting(tf, content, ds):
+    tf.clear()
+    for line in [l for l in content.split("\n") if l.strip()]:
+        clean = re.sub(r'^#+\s*','', line).strip()
+        if re.match(r'^(What is|Understanding|Key|Conclusion)', clean, re.IGNORECASE):
+            p = tf.add_paragraph(); p.text=clean
+            p.font.size=Pt(ds["heading_size"]); p.font.bold=True; p.space_after=Pt(12)
+            continue
+        if re.match(r'^[\-\*\•] ', clean):
+            p=tf.add_paragraph(); p.level=0
+            txt = clean[2:].strip()
+            for seg in split_formatted_runs(txt):
+                r=p.add_run(); r.text=seg["text"]; r.font.bold=seg["bold"]; r.font.italic=seg["italic"]
+            p.font.size=Pt(ds["body_size"]); continue
+        p=tf.add_paragraph()
+        for seg in split_formatted_runs(clean):
+            r=p.add_run(); r.text=seg["text"]; r.font.bold=seg["bold"]; r.font.italic=seg["italic"]
+        p.font.size=Pt(ds["body_size"]); p.space_after=Pt(6)
+
+def apply_design(slide, ds):
+    cols = ds["colors"]
+    for shp in slide.shapes:
+        if hasattr(shp, "text_frame"):
+            for p in shp.text_frame.paragraphs:
+                for r in p.runs:
+                    r.font.name = ds["font"]
+                    r.font.color.rgb = RGBColor.from_string(cols["text"][1:])
+                    if shp == getattr(slide.shapes, "title", None):
+                        r.font.color.rgb = RGBColor.from_string(cols["primary"][1:])
+                        r.font.size = Pt(ds["title_size"]); r.font.bold=True
+    if ds["set_background"]:
+        slide.background.fill.solid(); 
+        slide.background.fill.fore_color.rgb = RGBColor.from_string(cols["background"][1:])
+
+def enhance_slide(slide, prompt, ds):
+    orig = extract_slide_text(slide)
+    if not orig: return
+    frags = chunk_slide_text(orig, max_words=200)
+    enhanced_frags = []
+    for f in frags:
+        enhanced_frags.append(get_ai_response(f"\nImprove this:\n{f}\n\nInstructions: {prompt}"))
+    final = "\n".join(enhanced_frags)
+    title_sp = getattr(slide.shapes, "title", None)
+    for shp in list(slide.shapes):
+        if shp is not title_sp:
+            shp._element.getparent().remove(shp._element)
+    left, top = Inches(1), Inches(1.5 if title_sp else 0.5)
+    tb = slide.shapes.add_textbox(left, top, Inches(8), Inches(5))
+    tf = tb.text_frame; tf.word_wrap=True
+    apply_formatting(tf, final, ds)
+    apply_design(slide, ds)
+
+def process_presentation(uploaded, prompt, ds):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pptx") as tmp:
+        tmp.write(uploaded.read()); tmp_path=tmp.name
+    prs = Presentation(tmp_path)
+    for s in prs.slides:
+        enhance_slide(s, prompt, ds)
+    os.unlink(tmp_path)
+    return prs
+
+# ── UI ─────────────────────────────────────────────────────────
+st.title("Professional PPTX Enhancer")
+
+uploaded = st.file_uploader("Upload PPTX", type=["pptx"])
+user_prompt = st.text_area("Enhancement Instructions", placeholder="E.g. Improve clarity, add bullets", height=100)
+
+with st.expander("🎨 Design Settings", True):
+    tpl = st.selectbox("Template", list(TEMPLATES.keys()))
+    if tpl=="Custom":
+        font = st.selectbox("Font", list(FONT_CHOICES))
+        ts = st.slider("Title size",12,60,32)
+        hs = st.slider("Heading size",12,48,24)
+        bs = st.slider("Body size",8,36,18)
+        sb = st.checkbox("Apply Background",True)
+        cols = {k:st.color_picker(k.capitalize(),c) for k,c in TEMPLATES["Office Default"]["colors"].items()}
+    else:
+        t = TEMPLATES[tpl]
+        font, ts, hs, bs, sb = t["font"], *t["sizes"].values(), t["background"]
+        cols = t["colors"]
+
+ds = {
+    "font": FONT_CHOICES[font],
+    "colors": cols,
+    "title_size": ts,
+    "heading_size": hs,
+    "body_size": bs,
+    "set_background": sb
+}
+
+if st.button("Enhance Presentation") and uploaded and user_prompt:
+    out_prs = process_presentation(uploaded, user_prompt, ds)
+    buf = io.BytesIO(); out_prs.save(buf); buf.seek(0)
+    st.success("Done!")
+    st.download_button("Download Enhanced PPTX",
+                       data=buf,
+                       file_name="enhanced.pptx",
+                       mime="application/vnd.openxmlformats-officedocument.presentationml.presentation")