Update app.py

app.py

@@ -1,45 +1,52 @@
-# =========================== #
-#        AI Study Wizard      #
-# =========================== #
+# ============================== #
+#       AI Study Wizard v2       #
+# ============================== #
 
 import os
 import tempfile
 import random
-import requests
 import pytesseract
-from urllib.parse import quote_plus
+import requests
 
 from PIL import Image, ImageDraw, ImageFont
 from gtts import gTTS
 
-# MoviePy SAFE Imports for HuggingFace
+# Safe MoviePy Imports for HuggingFace
 from moviepy.video.VideoClip import ImageClip
 from moviepy.audio.io.AudioFileClip import AudioFileClip
 from moviepy.video.compositing.concatenate import concatenate_videoclips
 
+# PDF support
+from pdf2image import convert_from_path
+
 import gradio as gr
+from urllib.parse import quote_plus
 
-# =========================== #
-#  Load GROQ API Key (HF Secret)
-# =========================== #
-GROQ_API_KEY = os.getenv("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")
+
+# ==========================================================
+#  HARD-CODED GROQ API KEY (Replace with your real API key)
+# ==========================================================
+GROQ_API_KEY = "YOUR_GROQ_API_KEYgsk_GHftv8pvXaU7yeIltEaxWGdyb3FYlEDMH09TDRgJz5yzGQdVE7Rf"
 GROQ_MODEL = "llama-3.3-70b-versatile"
 
-# =========================== #
-#   Users Database / State
-# =========================== #
+
+# ==========================================================
+#  USERS DB / STATE
+# ==========================================================
 users_db = {}
 verification_codes = {}
 
-# =========================== #
-#   Signup / Verify / Login
-# =========================== #
+
+# ==========================================================
+#  SIGNUP / VERIFY / LOGIN SYSTEM
+# ==========================================================
 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,
@@ -74,18 +81,17 @@ def login_user(email, password):
         return False, "Account not verified."
 
     if user["password"] != password:
-        return False, "Incorrect password."
+        return False, "Wrong password."
 
     return True, f"Welcome {user['username']}!"
 
-# =========================== #
-#       Groq API
-# =========================== #
-def call_groq(prompt, max_tokens=500):
-    if not GROQ_API_KEY:
-        return "API key missing. Add GROQ_API_KEY in HuggingFace Secrets."
 
+# ==========================================================
+#  GROQ API CALL
+# ==========================================================
+def call_groq(prompt, max_tokens=600):
     url = "https://api.groq.com/openai/v1/chat/completions"
+
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
@@ -99,83 +105,72 @@ def call_groq(prompt, max_tokens=500):
 
     try:
         r = requests.post(url, json=payload, headers=headers)
-        data = r.json()
-        return data["choices"][0]["message"]["content"]
+        return r.json()["choices"][0]["message"]["content"]
     except Exception as e:
         return f"Groq API Error: {e}"
 
 
-def generate_mcqs_from_text(text, n=5):
-    prompt = f"Generate {n} MCQs with 4 options and correct answer:\n\n{text}"
-    return call_groq(prompt)
+def answer_text_query(q):
+    return call_groq(f"Answer very clearly:\n\n{q}")
 
 
-def answer_text_query(q):
-    prompt = f"Answer clearly:\n\n{q}"
-    return call_groq(prompt)
+def generate_mcqs_from_text(text, n=5):
+    return call_groq(f"Generate {n} MCQs with 4 options + correct answer:\n\n{text}")
 
-# =========================== #
-#   Video Generation Function
-# =========================== #
+
+# ==========================================================
+#  VIDEO GENERATION
+# ==========================================================
 def create_video_simple(text, tts_lang="en", preview=False):
-    try:
-        text = text.strip()
-        if not text:
-            return None
 
-        tmp = tempfile.gettempdir()
+    if not text.strip():
+        return None
 
-        # Split into slides
-        lines = [l.strip() for l in text.split("\n") if l.strip()]
-        if preview:
-            lines = lines[:1]
+    tmp = tempfile.gettempdir()
+    lines = [l.strip() for l in text.split("\n") if l.strip()]
+    if preview:
+        lines = lines[:1]
 
-        chunks = [lines[i:i+2] for i in range(0, len(lines), 2)]
-        clips = []
+    clips = []
 
-        for idx, chunk in enumerate(chunks):
-            slide_text = " ".join(chunk)
+    for i, line in enumerate(lines):
+        img_path = os.path.join(tmp, f"slide_{i}.png")
+        audio_path = os.path.join(tmp, f"audio_{i}.mp3")
 
-            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)
 
-            # Slide image
-            W, H = 480, 320
-            img = Image.new("RGB", (W, H), (255, 182, 193))
-            draw = ImageDraw.Draw(img)
+        try:
+            font = ImageFont.truetype("DejaVuSans-Bold.ttf", 22)
+        except:
+            font = ImageFont.load_default()
 
-            try:
-                font = ImageFont.truetype("DejaVuSans-Bold.ttf", 22)
-            except:
-                font = ImageFont.load_default()
+        draw.text((20, 20), line, fill="black", font=font)
+        img.save(img_path)
 
-            draw.text((20, 20), slide_text, fill="black", font=font)
-            img.save(img_path)
+        # Audio
+        gTTS(text=line, lang=tts_lang).save(audio_path)
 
-            # Audio
-            gTTS(text=slide_text, lang=tts_lang).save(audio_path)
+        duration = max(3, min(len(line) // 10 + 2, 8))
 
-            duration = max(3, min(len(slide_text) // 12 + 2, 7))
+        clip = ImageClip(img_path).set_duration(duration)
+        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)
+    out_path = os.path.join(tmp, "prev.mp4" if preview else "full.mp4")
+    final = concatenate_videoclips(clips, method="compose")
+    final.write_videofile(out_path, fps=24, codec="libx264", audio_codec="aac", verbose=False)
 
-        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 out_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 concise study notes:\n\n{text}")
+    notes = call_groq(f"Summarize into bullet study notes:\n\n{text}")
 
     img = Image.new("RGB", (480, 320), (144, 238, 144))
     draw = ImageDraw.Draw(img)
@@ -187,16 +182,33 @@ def generate_notes_flowchart(text):
 
     y = 20
     for line in notes.split("\n"):
-        if y > 290: break
+        if y > 290:
+            break
         draw.text((10, y), line, fill="black", font=font)
         y += 18
 
     return notes, img
 
 
-# =========================== #
-#          Links
-# =========================== #
+# ==========================================================
+#  PDF + IMAGE OCR
+# ==========================================================
+def extract_text_from_file(file):
+
+    if file.name.endswith(".pdf"):
+        pages = convert_from_path(file.name)
+        text = ""
+        for page in pages:
+            text += pytesseract.image_to_string(page) + "\n"
+        return text
+
+    else:
+        return pytesseract.image_to_string(Image.open(file.name))
+
+
+# ==========================================================
+#  LINKS
+# ==========================================================
 def get_youtube_link(q):
     return f"https://www.youtube.com/results?search_query={quote_plus(q)}"
 
@@ -206,39 +218,44 @@ def get_google_link(q):
 
 
 def get_links_clickable(topic):
-    if not topic.strip():
-        return "Enter a topic", ""
-
     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")
+# ==========================================================
+#  UI — GRADIENT BACKGROUND (Green + Light Pink)
+# ==========================================================
+custom_css = """
+body {
+    background: linear-gradient(to bottom right, #90ee90, #ffb6c1);
+}
+"""
+
+
+# ==========================================================
+#  GRADIO UI
+# ==========================================================
+with gr.Blocks(css=custom_css) as demo:
 
-    # --------- Signup --------- #
+    gr.Markdown("## 🌸 **AI Study Wizard** — Pink + Green Themed Student Assistant")
+
+    # ---------------- SIGN UP ---------------- #
     with gr.Tab("Sign Up"):
         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]
-        )
+        su_msg = gr.Textbox(label="Message", interactive=False)
+        su_code = gr.Textbox(label="Your Verification Code", interactive=False)
 
-    # --------- Verify --------- #
-    with gr.Tab("Verify Account"):
+        su_btn.click(signup_user, inputs=[su_name, su_email, su_pass, su_contact],
+                     outputs=[su_msg, su_code])
+
+    # ---------------- VERIFY ---------------- #
+    with gr.Tab("Verify"):
         v_email = gr.Textbox(label="Email")
         v_code = gr.Textbox(label="Verification Code")
         v_btn = gr.Button("Verify")
@@ -246,66 +263,51 @@ with gr.Blocks() as demo:
 
         v_btn.click(verify_user, inputs=[v_email, v_code], outputs=v_msg)
 
-    # --------- Login --------- #
+    # ---------------- LOGIN ---------------- #
     with gr.Tab("Login"):
         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):
+        def login_wrap(e, p):
             ok, msg = login_user(e, p)
             return msg
 
-        li_btn.click(login_wrapper, inputs=[li_email, li_pass], outputs=li_msg)
+        li_btn.click(login_wrap, inputs=[li_email, li_pass], outputs=li_msg)
+
+    # ---------------- QUESTION ---------------- #
+    with gr.Tab("Ask Question"):
+        q_in = gr.Textbox(label="Ask Any Question")
+        a_out = gr.Textbox(label="Answer")
+        m_out = gr.Textbox(label="MCQs")
+        pv = gr.Video(label="Preview")
+        fv = gr.Video(label="Full Video")
+        ask_btn = gr.Button("Generate")
 
-    # --------- 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")
+        def run_question(q):
+            ans = answer_text_query(q)
+            mcqs = generate_mcqs_from_text(ans)
+            return ans, mcqs, create_video_simple(ans, preview=True), create_video_simple(ans)
 
-        prev_vid = gr.Video(label="Preview Video")
-        full_vid = gr.Video(label="Full Video")
+        ask_btn.click(run_question, inputs=q_in, outputs=[a_out, m_out, pv, fv])
 
-        ask_btn = gr.Button("Generate")
+    # ---------------- UPLOAD ---------------- #
+    with gr.Tab("Upload Document (PDF/Image)"):
+
+        f_in = gr.File()
+        options = gr.CheckboxGroup(["Extracted Text", "MCQs", "Video", "Notes"])
+        lang = gr.Dropdown(["en", "ur", "hi"], value="en", label="Video Language")
+
+        t_out = gr.Textbox(label="Extracted Text", lines=8)
+        mc_out = gr.Textbox(label="MCQs", lines=8)
+        pv2 = gr.Video(label="Preview Video")
+        fv2 = gr.Video(label="Full Video")
+        notes_out = gr.Textbox(label="Notes")
+        flow_out = gr.Image(label="Flowchart")
 
-        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(file.name))
-            except:
-                pass
+        def process_file(file, opts, lg):
+            text = extract_text_from_file(file)
 
             mcq = ""
             prev = None
@@ -317,8 +319,8 @@ with gr.Blocks() as demo:
                 mcq = generate_mcqs_from_text(text)
 
             if "Video" in opts:
-                prev = create_video_simple(text, lang, preview=True)
-                full = create_video_simple(text, lang, preview=False)
+                prev = create_video_simple(text, lg, preview=True)
+                full = create_video_simple(text, lg)
 
             if "Notes" in opts:
                 notes, flow = generate_notes_flowchart(text)
@@ -326,13 +328,10 @@ with gr.Blocks() as demo:
             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]
-        )
+        btn.click(process_file, inputs=[f_in, options, lang],
+                  outputs=[t_out, mc_out, pv2, fv2, notes_out, flow_out])
 
-    # --------- Links --------- #
+    # ---------------- LINKS ---------------- #
     with gr.Tab("Search Links"):
         topic = gr.Textbox(label="Topic")
         yt = gr.Markdown()