Update app.py

app.py

@@ -1,5 +1,5 @@
 # ============================== #
-#       AI Study Wizard v2       #
+#       AI Study Wizard v3       #
 # ============================== #
 
 import os
@@ -11,7 +11,7 @@ import requests
 from PIL import Image, ImageDraw, ImageFont
 from gtts import gTTS
 
-# Safe MoviePy Imports for HuggingFace
+# Safe MoviePy Imports
 from moviepy.video.VideoClip import ImageClip
 from moviepy.audio.io.AudioFileClip import AudioFileClip
 from moviepy.video.compositing.concatenate import concatenate_videoclips
@@ -22,23 +22,20 @@ from pdf2image import convert_from_path
 import gradio as gr
 from urllib.parse import quote_plus
 
-
 # ==========================================================
-#  HARD-CODED GROQ API KEY (Replace with your real API key)
+#  HARD-CODED GROQ API KEY (Replace with your key)
 # ==========================================================
-GROQ_API_KEY = "YOUR_GROQ_API_KEYgsk_GHftv8pvXaU7yeIltEaxWGdyb3FYlEDMH09TDRgJz5yzGQdVE7Rf"
+GROQ_API_KEY = "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"
 GROQ_MODEL = "llama-3.3-70b-versatile"
 
-
 # ==========================================================
 #  USERS DB / STATE
 # ==========================================================
 users_db = {}
 verification_codes = {}
 
-
 # ==========================================================
-#  SIGNUP / VERIFY / LOGIN SYSTEM
+#  SIGNUP / VERIFY / LOGIN
 # ==========================================================
 def signup_user(username, email, password, contact):
     if email in users_db:
@@ -46,299 +43,210 @@ def signup_user(username, email, password, contact):
 
     code = str(random.randint(100000, 999999))
     verification_codes[email] = code
-
-    users_db[email] = {
-        "username": username,
-        "password": password,
-        "contact": contact,
-        "verified": False
-    }
-
+    users_db[email] = {"username": username, "password": password, "contact": contact, "verified": False}
     return "Signup successful! Your verification code:", code
 
-
 def verify_user(email, code):
-    if email not in users_db:
-        return "Email not found."
-
-    if users_db[email]["verified"]:
-        return "Already verified."
-
+    if email not in users_db: return "Email not found."
+    if users_db[email]["verified"]: return "Already verified."
     if verification_codes.get(email) == code:
         users_db[email]["verified"] = True
         verification_codes.pop(email)
         return "Verification successful!"
-
     return "Invalid verification code."
 
-
 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, "Wrong password."
-
+    if not user: return False, "User not found."
+    if not user["verified"]: return False, "Account not verified."
+    if user["password"] != password: return False, "Wrong password."
     return True, f"Welcome {user['username']}!"
 
-
 # ==========================================================
 #  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"
-    }
-
-    payload = {
-        "model": GROQ_MODEL,
-        "messages": [{"role": "user", "content": prompt}],
-        "max_tokens": max_tokens
-    }
-
+    headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
+    payload = {"model": GROQ_MODEL, "messages":[{"role":"user","content":prompt}], "max_tokens": max_tokens}
     try:
         r = requests.post(url, json=payload, headers=headers)
         return r.json()["choices"][0]["message"]["content"]
     except Exception as e:
         return f"Groq API Error: {e}"
 
-
+# ==========================================================
+#  ANSWER & MCQs
+# ==========================================================
 def answer_text_query(q):
-    return call_groq(f"Answer very clearly:\n\n{q}")
-
+    q = q.strip()
+    if len(q.split()) < 5:
+        prompt = f"Explain the following clearly and concisely for a beginner:\n\n{q}"
+    else:
+        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
 # ==========================================================
 def create_video_simple(text, tts_lang="en", preview=False):
-
-    if not text.strip():
-        return None
-
+    if not text.strip(): return None
     tmp = tempfile.gettempdir()
     lines = [l.strip() for l in text.split("\n") if l.strip()]
-    if preview:
-        lines = lines[:1]
-
+    if preview: lines = lines[:1]
     clips = []
-
     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")
-
-        # Slide image
-        W, H = 480, 320
-        img = Image.new("RGB", (W, H), (255, 182, 193))
+        # Image
+        W, H = 480, 360
+        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()
-
-        draw.text((20, 20), line, fill="black", font=font)
+        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)
-
         # Audio
         gTTS(text=line, lang=tts_lang).save(audio_path)
-
-        duration = max(3, min(len(line) // 10 + 2, 8))
-
-        clip = ImageClip(img_path).set_duration(duration)
-        clip = clip.set_audio(AudioFileClip(audio_path))
+        duration = max(3, min(len(line)//10+2, 8))
+        clip = ImageClip(img_path).set_duration(duration).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)
-
     return out_path
 
-
 # ==========================================================
 #  NOTES + FLOWCHART
 # ==========================================================
 def generate_notes_flowchart(text):
     notes = call_groq(f"Summarize into bullet study notes:\n\n{text}")
-
-    img = Image.new("RGB", (480, 320), (144, 238, 144))
+    img = Image.new("RGB", (480,360), (144,238,144))
     draw = ImageDraw.Draw(img)
-
-    try:
-        font = ImageFont.truetype("DejaVuSans.ttf", 16)
-    except:
-        font = ImageFont.load_default()
-
+    try: font = ImageFont.truetype("DejaVuSans.ttf", 16)
+    except: font = ImageFont.load_default()
     y = 20
     for line in notes.split("\n"):
-        if y > 290:
-            break
-        draw.text((10, y), line, fill="black", font=font)
-        y += 18
-
-    return notes, img
-
+        if y>330: break
+        draw.text((10,y), line, fill="black", font=font)
+        y+=18
+    return notes,img
 
 # ==========================================================
-#  PDF + IMAGE OCR
+#  FILE OCR (PDF + IMAGE)
 # ==========================================================
 def extract_text_from_file(file):
-
     if file.name.endswith(".pdf"):
         pages = convert_from_path(file.name)
-        text = ""
+        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)}"
-
-
-def get_google_link(q):
-    return f"https://www.google.com/search?q={quote_plus(q)}"
-
-
+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)}"
 def get_links_clickable(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
-
+    return yt,gg
 
 # ==========================================================
-#  UI — GRADIENT BACKGROUND (Green + Light Pink)
+#  GRADIENT BACKGROUND
 # ==========================================================
 custom_css = """
-body {
-    background: linear-gradient(to bottom right, #90ee90, #ffb6c1);
-}
+body {background: linear-gradient(to bottom right, #90ee90, #ffb6c1);}
 """
 
-
 # ==========================================================
-#  GRADIO UI
+#  GRADIO APP
 # ==========================================================
 with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("## 🌸 **AI Study Wizard** — Pink + Green Themed")
 
-    gr.Markdown("## 🌸 **AI Study Wizard** — Pink + Green Themed Student Assistant")
-
-    # ---------------- SIGN UP ---------------- #
+    # -------- SIGNUP -------- #
     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")
-
-        su_msg = gr.Textbox(label="Message", interactive=False)
-        su_code = gr.Textbox(label="Your Verification Code", interactive=False)
-
-        su_btn.click(signup_user, inputs=[su_name, su_email, su_pass, su_contact],
-                     outputs=[su_msg, su_code])
-
-    # ---------------- 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")
+        su_msg=gr.Textbox(label="Message", interactive=False)
+        su_code=gr.Textbox(label="Verification Code", interactive=False)
+        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")
-        v_msg = gr.Textbox(label="Status", interactive=False)
+        v_email=gr.Textbox(label="Email")
+        v_code=gr.Textbox(label="Verification Code")
+        v_btn=gr.Button("Verify")
+        v_msg=gr.Textbox(label="Status", interactive=False)
+        v_btn.click(verify_user, inputs=[v_email,v_code], outputs=v_msg)
 
-        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_wrap(e, p):
-            ok, msg = login_user(e, p)
-            return msg
-
-        li_btn.click(login_wrap, inputs=[li_email, li_pass], outputs=li_msg)
-
-    # ---------------- QUESTION ---------------- #
+        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_wrap(e,p): ok,msg=login_user(e,p); return 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")
-
+        q_in=gr.Textbox(label="Ask Any Question")
+        a_out=gr.Textbox(label="Answer", lines=12)
+        m_out=gr.Textbox(label="MCQs", lines=12)
+        pv=gr.Video(label="Preview Video", width=640, height=360)
+        fv=gr.Video(label="Full Video", width=640, height=360)
+        ask_btn=gr.Button("Generate")
         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)
+            ans=answer_text_query(q)
+            mcqs=generate_mcqs_from_text(ans)
+            return ans,mcqs,create_video_simple(ans,preview=True),create_video_simple(ans)
+        ask_btn.click(run_question, inputs=q_in, outputs=[a_out,m_out,pv,fv])
 
-        ask_btn.click(run_question, inputs=q_in, outputs=[a_out, m_out, pv, fv])
-
-    # ---------------- UPLOAD ---------------- #
+    # -------- 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 process_file(file, opts, lg):
-            text = extract_text_from_file(file)
-
-            mcq = ""
-            prev = None
-            full = None
-            notes = ""
-            flow = None
-
-            if "MCQs" in opts:
-                mcq = generate_mcqs_from_text(text)
-
-            if "Video" in opts:
-                prev = create_video_simple(text, lg, preview=True)
-                full = create_video_simple(text, lg)
-
-            if "Notes" in opts:
-                notes, flow = generate_notes_flowchart(text)
-
-            return text, mcq, prev, full, notes, flow
-
-        btn = gr.Button("Process")
-        btn.click(process_file, inputs=[f_in, options, lang],
-                  outputs=[t_out, mc_out, pv2, fv2, notes_out, flow_out])
-
-    # ---------------- LINKS ---------------- #
+        f_in=gr.File()
+        opts=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=12)
+        mc_out=gr.Textbox(label="MCQs", lines=12)
+        pv2=gr.Video(label="Preview Video", width=640, height=360)
+        fv2=gr.Video(label="Full Video", width=640, height=360)
+        notes_out=gr.Textbox(label="Notes", lines=12)
+        flow_out=gr.Image(label="Flowchart", type="pil")
+        def process_file(file, options, lg):
+            text=extract_text_from_file(file)
+            mcq=""
+            prev=None
+            full=None
+            notes=""
+            flow=None
+            if "MCQs" in options: mcq=generate_mcqs_from_text(text)
+            if "Video" in options:
+                prev=create_video_simple(text,lg,preview=True)
+                full=create_video_simple(text,lg)
+            if "Notes" in options:
+                notes,flow=generate_notes_flowchart(text)
+            return text,mcq,prev,full,notes,flow
+        btn=gr.Button("Process")
+        btn.click(process_file, inputs=[f_in,opts,lang], outputs=[t_out,mc_out,pv2,fv2,notes_out,flow_out])
+
+    # -------- LINKS -------- #
     with gr.Tab("Search Links"):
-        topic = gr.Textbox(label="Topic")
-        yt = gr.Markdown()
-        gg = gr.Markdown()
-        link_btn = gr.Button("Get Links")
-
-        link_btn.click(get_links_clickable, inputs=topic, outputs=[yt, gg])
-
+        topic=gr.Textbox(label="Topic")
+        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()