Update app.py

app.py

@@ -1,40 +1,45 @@
+# =========================== #
+#        AI Study Wizard      #
+# =========================== #
+
 import os
 import tempfile
 import random
 import requests
+import pytesseract
 from urllib.parse import quote_plus
+
 from PIL import Image, ImageDraw, ImageFont
-import pytesseract
 from gtts import gTTS
-from moviepy.editor import ImageClip, AudioFileClip, concatenate_videoclips
-import gradio as gr
 
-# ===========================
-# Environment Variables
-# ===========================
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")
-GROQ_MODEL = "llama-3.3-70b-versatile"
+# MoviePy SAFE Imports for HuggingFace
+from moviepy.video.VideoClip import ImageClip
+from moviepy.audio.io.AudioFileClip import AudioFileClip
+from moviepy.video.compositing.concatenate import concatenate_videoclips
 
-if not GROQ_API_KEY:
-    print("⚠️ WARNING: GROQ_API_KEY is missing. Add it in HuggingFace Secrets!")
+import gradio as gr
 
+# =========================== #
+#  Load GROQ API Key (HF Secret)
+# =========================== #
+GROQ_API_KEY = os.getenv("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")
+GROQ_MODEL = "llama-3.3-70b-versatile"
 
-# ===========================
-# Users Database
-# ===========================
+# =========================== #
+#   Users Database / State
+# =========================== #
 users_db = {}
 verification_codes = {}
 
-
-# ===========================
-# Authentication Functions
-# ===========================
+# =========================== #
+#   Signup / Verify / Login
+# =========================== #
 def signup_user(username, email, password, contact):
     if email in users_db:
         return "Email already registered.", ""
+
     code = str(random.randint(100000, 999999))
     verification_codes[email] = code
-
     users_db[email] = {
         "username": username,
         "password": password,
@@ -42,12 +47,13 @@ def signup_user(username, email, password, contact):
         "verified": False
     }
 
-    return "Signup successful! Verification code:", code
+    return "Signup successful! Your verification code:", code
 
 
 def verify_user(email, code):
     if email not in users_db:
-        return "Email not registered."
+        return "Email not found."
+
     if users_db[email]["verified"]:
         return "Already verified."
 
@@ -55,6 +61,7 @@ def verify_user(email, code):
         users_db[email]["verified"] = True
         verification_codes.pop(email)
         return "Verification successful!"
+
     return "Invalid verification code."
 
 
@@ -62,22 +69,27 @@ def login_user(email, password):
     user = users_db.get(email)
     if not user:
         return False, "User not found."
+
     if not user["verified"]:
         return False, "Account not verified."
+
     if user["password"] != password:
         return False, "Incorrect password."
-    return True, f"Welcome {user['username']}!"
 
+    return True, f"Welcome {user['username']}!"
 
-# ===========================
-# Groq API Call
-# ===========================
+# =========================== #
+#       Groq API
+# =========================== #
 def call_groq(prompt, max_tokens=500):
     if not GROQ_API_KEY:
-        return "❌ Missing GROQ_API_KEY in environment variables."
+        return "API key missing. Add GROQ_API_KEY in HuggingFace Secrets."
 
     url = "https://api.groq.com/openai/v1/chat/completions"
-    headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
 
     payload = {
         "model": GROQ_MODEL,
@@ -86,184 +98,248 @@ def call_groq(prompt, max_tokens=500):
     }
 
     try:
-        r = requests.post(url, json=payload, headers=headers, timeout=20)
-        if r.status_code == 200:
-            return r.json()["choices"][0]["message"]["content"]
-        return f"API Error {r.status_code}: {r.text}"
+        r = requests.post(url, json=payload, headers=headers)
+        data = r.json()
+        return data["choices"][0]["message"]["content"]
     except Exception as e:
-        return f"API Exception: {e}"
+        return f"Groq API Error: {e}"
 
 
-# ===========================
-# AI Helper Functions
-# ===========================
 def generate_mcqs_from_text(text, n=5):
-    prompt = f"Create {n} MCQs with 4 options and answers:\n\n{text}"
+    prompt = f"Generate {n} MCQs with 4 options and correct answer:\n\n{text}"
     return call_groq(prompt)
 
 
-def answer_text_query(query):
-    prompt = f"Answer clearly:\n\n{query}"
+def answer_text_query(q):
+    prompt = f"Answer clearly:\n\n{q}"
     return call_groq(prompt)
 
-
-# ===========================
-# Video Generator
-# ===========================
+# =========================== #
+#   Video Generation Function
+# =========================== #
 def create_video_simple(text, tts_lang="en", preview=False):
-    if not text.strip():
-        return None
+    try:
+        text = text.strip()
+        if not text:
+            return None
 
-    tmp = tempfile.gettempdir()
+        tmp = tempfile.gettempdir()
 
-    lines = [l.strip() for l in text.split(".") if l.strip()]
-    if preview:
-        lines = lines[:1]  # one slide preview
+        # Split into slides
+        lines = [l.strip() for l in text.split("\n") if l.strip()]
+        if preview:
+            lines = lines[:1]
 
-    clips = []
+        chunks = [lines[i:i+2] for i in range(0, len(lines), 2)]
+        clips = []
 
-    for i, chunk in enumerate(lines):
-        slide_path = os.path.join(tmp, f"slide_{i}.png")
+        for idx, chunk in enumerate(chunks):
+            slide_text = " ".join(chunk)
 
-        # Create slide
-        img = Image.new("RGB", (480, 320), (255, 182, 193))
-        draw = ImageDraw.Draw(img)
-        font = ImageFont.load_default()
+            img_path = os.path.join(tmp, f"slide_{idx}.png")
+            audio_path = os.path.join(tmp, f"audio_{idx}.mp3")
+
+            # Slide image
+            W, H = 480, 320
+            img = Image.new("RGB", (W, H), (255, 182, 193))
+            draw = ImageDraw.Draw(img)
 
-        draw.text((20, 20), "AI Generated Slide", fill="blue", font=font)
-        draw.text((20, 80), chunk[:220], fill="black", font=font)
+            try:
+                font = ImageFont.truetype("DejaVuSans-Bold.ttf", 22)
+            except:
+                font = ImageFont.load_default()
+
+            draw.text((20, 20), slide_text, fill="black", font=font)
+            img.save(img_path)
 
-        img.save(slide_path)
+            # Audio
+            gTTS(text=slide_text, lang=tts_lang).save(audio_path)
 
-        # Audio
-        audio_path = os.path.join(tmp, f"audio_{i}.mp3")
-        gTTS(text=chunk, lang=tts_lang).save(audio_path)
+            duration = max(3, min(len(slide_text) // 12 + 2, 7))
 
-        # Clip
-        clip = ImageClip(slide_path).set_duration(4)
-        clip = clip.set_audio(AudioFileClip(audio_path))
-        clips.append(clip)
+            clip = ImageClip(img_path).set_duration(duration)
+            clip = clip.set_audio(AudioFileClip(audio_path))
+            clips.append(clip)
 
-    output_path = os.path.join(tmp, f"video_preview.mp4" if preview else "video_full.mp4")
-    video = concatenate_videoclips(clips)
-    video.write_videofile(output_path, fps=24, codec="libx264", audio_codec="aac")
+        out_path = os.path.join(tmp, "preview.mp4" if preview else "video.mp4")
+        final = concatenate_videoclips(clips, method="compose")
+        final.write_videofile(out_path, fps=24, codec="libx264", audio_codec="aac", verbose=False)
 
-    return output_path
+        return out_path
 
+    except Exception as e:
+        print("Video Error:", e)
+        return None
 
-# ===========================
-# Notes + Flowchart
-# ===========================
+# =========================== #
+#     Notes + Flowchart
+# =========================== #
 def generate_notes_flowchart(text):
-    notes = call_groq(f"Summarize into clean notes:\n\n{text}")
-    img = Image.new("RGB", (500, 350), (180, 250, 180))
+    notes = call_groq(f"Summarize into concise study notes:\n\n{text}")
+
+    img = Image.new("RGB", (480, 320), (144, 238, 144))
     draw = ImageDraw.Draw(img)
-    font = ImageFont.load_default()
+
+    try:
+        font = ImageFont.truetype("DejaVuSans.ttf", 16)
+    except:
+        font = ImageFont.load_default()
 
     y = 20
     for line in notes.split("\n"):
-        draw.text((10, y), line[:60], fill="black", font=font)
+        if y > 290: break
+        draw.text((10, y), line, fill="black", font=font)
         y += 18
-        if y > 300:
-            break
+
     return notes, img
 
 
-# ===========================
-# Quick Search Links
-# ===========================
-def get_links(topic):
-    yt = f"https://www.youtube.com/results?search_query={quote_plus(topic)}"
-    gg = f"https://www.google.com/search?q={quote_plus(topic)}"
-    return yt, gg
+# =========================== #
+#          Links
+# =========================== #
+def get_youtube_link(q):
+    return f"https://www.youtube.com/results?search_query={quote_plus(q)}"
+
 
+def get_google_link(q):
+    return f"https://www.google.com/search?q={quote_plus(q)}"
 
-# ===========================
-# UI
-# ===========================
-with gr.Blocks(title="AI Study Wizard") as demo:
 
-    gr.Markdown("# 🎓 AI Study Wizard — Full Assistant")
+def get_links_clickable(topic):
+    if not topic.strip():
+        return "Enter a topic", ""
 
-    # Signup
+    yt = "\n".join([f"[YouTube {i+1}]({get_youtube_link(topic)})" for i in range(3)])
+    gg = "\n".join([f"[Google {i+1}]({get_google_link(topic)})" for i in range(3)])
+    return yt, gg
+
+# =========================== #
+#           UI
+# =========================== #
+with gr.Blocks() as demo:
+
+    gr.Markdown("## 🎓 AI Study Wizard — Smart Student Assistant")
+
+    # --------- Signup --------- #
     with gr.Tab("Sign Up"):
-        name = gr.Textbox(label="Name")
-        email = gr.Textbox(label="Email")
-        pwd = gr.Textbox(label="Password", type="password")
-        contact = gr.Textbox(label="Contact")
-        btn = gr.Button("Create Account")
-        msg = gr.Textbox(label="Message")
-        code_out = gr.Textbox(label="Verification Code (demo)")
-        btn.click(signup_user, [name, email, pwd, contact], [msg, code_out])
-
-    # Verify
-    with gr.Tab("Verify"):
+        su_name = gr.Textbox(label="Username")
+        su_email = gr.Textbox(label="Email")
+        su_pass = gr.Textbox(label="Password", type="password")
+        su_contact = gr.Textbox(label="Contact Number")
+
+        su_btn = gr.Button("Sign Up")
+        msg = gr.Textbox(label="Message", interactive=False)
+        code_show = gr.Textbox(label="Verification Code", interactive=False)
+
+        su_btn.click(
+            signup_user,
+            inputs=[su_name, su_email, su_pass, su_contact],
+            outputs=[msg, code_show]
+        )
+
+    # --------- Verify --------- #
+    with gr.Tab("Verify Account"):
         v_email = gr.Textbox(label="Email")
-        v_code = gr.Textbox(label="Code")
+        v_code = gr.Textbox(label="Verification Code")
         v_btn = gr.Button("Verify")
-        v_msg = gr.Textbox(label="Status")
-        v_btn.click(verify_user, [v_email, v_code], v_msg)
+        v_msg = gr.Textbox(label="Status", interactive=False)
+
+        v_btn.click(verify_user, inputs=[v_email, v_code], outputs=v_msg)
 
-    # Login
+    # --------- Login --------- #
     with gr.Tab("Login"):
-        l_email = gr.Textbox(label="Email")
-        l_pass = gr.Textbox(label="Password", type="password")
-        l_btn = gr.Button("Login")
-        l_msg = gr.Textbox(label="Message")
-        l_btn.click(lambda e, p: login_user(e, p), [l_email, l_pass], l_msg)
-
-    # Ask Question
-    with gr.Tab("Ask AI"):
-        q = gr.Textbox(label="Your Question")
-        ans = gr.Textbox(label="Answer", lines=8)
-        mcq = gr.Textbox(label="MCQs", lines=8)
-        pv = gr.Video(label="Preview Video")
-        fv = gr.Video(label="Full Video")
-        b = gr.Button("Generate")
-
-        def process_query(q):
-            a = answer_text_query(q)
-            m = generate_mcqs_from_text(a)
-            prev = create_video_simple(a, preview=True)
-            full = create_video_simple(a, preview=False)
-            return a, m, prev, full
-
-        b.click(process_query, q, [ans, mcq, pv, fv])
-
-    # Upload Notes
-    with gr.Tab("Upload Image"):
-        file = gr.File(label="Upload Image")
-        lang = gr.Dropdown(["en", "ur", "hi"], value="en")
-        out_text = gr.Textbox(label="Extracted Text", lines=6)
-        out_mcq = gr.Textbox(label="MCQs", lines=6)
-        out_notes = gr.Textbox(label="Notes", lines=6)
-        out_flow = gr.Image(label="Flowchart")
-        out_prev = gr.Video(label="Preview Video")
-        out_full = gr.Video(label="Full Video")
-
-        def process_upload(f, language):
+        li_email = gr.Textbox(label="Email")
+        li_pass = gr.Textbox(label="Password", type="password")
+        li_btn = gr.Button("Login")
+        li_msg = gr.Textbox(label="Login Status")
+
+        def login_wrapper(e, p):
+            ok, msg = login_user(e, p)
+            return msg
+
+        li_btn.click(login_wrapper, inputs=[li_email, li_pass], outputs=li_msg)
+
+    # --------- Question --------- #
+    with gr.Tab("Ask a Question"):
+        q_in = gr.Textbox(label="Your Question")
+        ans_out = gr.Textbox(label="Answer")
+        mcq_out = gr.Textbox(label="MCQs")
+
+        prev_vid = gr.Video(label="Preview Video")
+        full_vid = gr.Video(label="Full Video")
+
+        ask_btn = gr.Button("Generate")
+
+        def ask_system(q):
+            answer = answer_text_query(q)
+            mcqs = generate_mcqs_from_text(answer)
+            return (
+                answer,
+                mcqs,
+                create_video_simple(answer, preview=True),
+                create_video_simple(answer, preview=False)
+            )
+
+        ask_btn.click(ask_system, inputs=q_in,
+                      outputs=[ans_out, mcq_out, prev_vid, full_vid])
+
+    # --------- Upload --------- #
+    with gr.Tab("Upload Document"):
+        file_in = gr.File()
+        options = gr.CheckboxGroup(
+            ["Extracted Text", "MCQs", "Video", "Notes"]
+        )
+        video_lang = gr.Dropdown(
+            ["en", "ur", "hi"], value="en", label="Video Language"
+        )
+
+        o_text = gr.Textbox(label="Extracted Text", lines=8)
+        o_mcq = gr.Textbox(label="MCQs", lines=8)
+        o_prev = gr.Video(label="Preview Video")
+        o_vid = gr.Video(label="Full Video")
+        o_notes = gr.Textbox(label="Notes")
+        o_flow = gr.Image(label="Flowchart")
+
+        def process(file, opts, lang):
+            text = ""
             try:
-                text = pytesseract.image_to_string(Image.open(f.name))
+                text = pytesseract.image_to_string(Image.open(file.name))
             except:
-                text = ""
+                pass
 
-            mcqs = generate_mcqs_from_text(text) if text else ""
-            notes, img = generate_notes_flowchart(text) if text else ("", None)
-            prev = create_video_simple(text, tts_lang=language, preview=True) if text else None
-            full = create_video_simple(text, tts_lang=language, preview=False) if text else None
+            mcq = ""
+            prev = None
+            full = None
+            notes = ""
+            flow = None
 
-            return text, mcqs, notes, img, prev, full
+            if "MCQs" in opts:
+                mcq = generate_mcqs_from_text(text)
 
-        btn = gr.Button("Process")
-        btn.click(process_upload, [file, lang],
-                  [out_text, out_mcq, out_notes, out_flow, out_prev, out_full])
+            if "Video" in opts:
+                prev = create_video_simple(text, lang, preview=True)
+                full = create_video_simple(text, lang, preview=False)
 
-    # Search Links
-    with gr.Tab("Search"):
+            if "Notes" in opts:
+                notes, flow = generate_notes_flowchart(text)
+
+            return text, mcq, prev, full, notes, flow
+
+        btn = gr.Button("Process")
+        btn.click(
+            process,
+            inputs=[file_in, options, video_lang],
+            outputs=[o_text, o_mcq, o_prev, o_vid, o_notes, o_flow]
+        )
+
+    # --------- Links --------- #
+    with gr.Tab("Search Links"):
         topic = gr.Textbox(label="Topic")
-        yt = gr.Textbox(label="YouTube", interactive=False)
-        gg = gr.Textbox(label="Google", interactive=False)
-        btn = gr.Button("Search")
-        btn.click(get_links, topic, [yt, gg])
+        yt = gr.Markdown()
+        gg = gr.Markdown()
+        link_btn = gr.Button("Get Links")
+
+        link_btn.click(get_links_clickable, inputs=topic, outputs=[yt, gg])
+
 
 demo.launch()