Update app.py

app.py

@@ -1,272 +1,269 @@
-# =========================== #
-#       Install Packages      #
-# =========================== #
-import os, tempfile, random
+import os
+import tempfile
+import random
+import requests
+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
-from urllib.parse import quote_plus
-import requests
 
-# =========================== #
-# Users DB & States           #
-# =========================== #
+# ===========================
+# Environment Variables
+# ===========================
+GROQ_API_KEY = os.getenv("GROQ_API_KEY", "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")
+GROQ_MODEL = "llama-3.3-70b-versatile"
+
+if not GROQ_API_KEY:
+    print("⚠️ WARNING: GROQ_API_KEY is missing. Add it in HuggingFace Secrets!")
+
+
+# ===========================
+# Users Database
+# ===========================
 users_db = {}
 verification_codes = {}
 
-# =========================== #
-# Signup & Verification       #
-# =========================== #
-def signup_user(username,email,password,contact):
+
+# ===========================
+# Authentication Functions
+# ===========================
+def signup_user(username, email, password, contact):
     if email in users_db:
-        return "Email already registered. Please login.", ""
-    code = str(random.randint(100000,999999))
+        return "Email already registered.", ""
+    code = str(random.randint(100000, 999999))
     verification_codes[email] = code
-    users_db[email] = {"username":username,"password":password,"contact":contact,"verified":False}
-    return f"Signup successful! Your verification code (demo) is:", code
 
-def verify_user(email,code):
+    users_db[email] = {
+        "username": username,
+        "password": password,
+        "contact": contact,
+        "verified": False
+    }
+
+    return "Signup successful! Verification code:", code
+
+
+def verify_user(email, code):
     if email not in users_db:
-        return "Email not found."
+        return "Email not registered."
     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! You can now login."
+        return "Verification successful!"
     return "Invalid verification code."
 
-def login_user(email,password):
+
+def login_user(email, password):
     user = users_db.get(email)
-    if not user: return False,"User not found."
-    if not user["verified"]: return False,"Not verified."
-    if user["password"] != password: return False,"Incorrect password."
-    return True,f"Welcome {user['username']}!"
-
-# =========================== #
-# Groq API Functions          #
-# =========================== #
-GROQ_API_KEY = "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"
-GROQ_MODEL = "llama-3.3-70b-versatile"
+    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']}!"
+
+
+# ===========================
+# Groq API Call
+# ===========================
+def call_groq(prompt, max_tokens=500):
+    if not GROQ_API_KEY:
+        return "❌ Missing GROQ_API_KEY in environment variables."
 
-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"}
+    headers = {"Authorization": f"Bearer {GROQ_API_KEY}", "Content-Type": "application/json"}
+
     payload = {
-        "model":GROQ_MODEL,
-        "messages":[{"role":"user","content":prompt}],
-        "max_tokens":max_tokens
+        "model": GROQ_MODEL,
+        "messages": [{"role": "user", "content": prompt}],
+        "max_tokens": max_tokens
     }
+
     try:
-        r = requests.post(url,json=payload,headers=headers,timeout=30)
+        r = requests.post(url, json=payload, headers=headers, timeout=20)
         if r.status_code == 200:
-            return r.json().get("choices",[{}])[0].get("message",{}).get("content","")
-        return f"[Model error {r.status_code}] {r.text}"
+            return r.json()["choices"][0]["message"]["content"]
+        return f"API Error {r.status_code}: {r.text}"
     except Exception as e:
-        return f"[API error] {e}"
+        return f"API Exception: {e}"
 
-def generate_mcqs_from_text(text,n=5):
-    if not text.strip(): return "No text provided."
-    prompt=f"Create {n} MCQs with 4 options and correct answer:\n\n{text}"
+
+# ===========================
+# AI Helper Functions
+# ===========================
+def generate_mcqs_from_text(text, n=5):
+    prompt = f"Create {n} MCQs with 4 options and answers:\n\n{text}"
     return call_groq(prompt)
 
+
 def answer_text_query(query):
-    if not query.strip(): return "No question provided."
-    prompt=f"Answer clearly and concisely:\n\n{query}"
+    prompt = f"Answer clearly:\n\n{query}"
     return call_groq(prompt)
 
-# =========================== #
-# Video Generation            #
-# =========================== #
-def create_video_simple(text,tts_lang="en",preview=False):
-    if not text.strip(): return None
-    tmp=tempfile.gettempdir()
-    lines=[l.strip() for l in text.splitlines() if l.strip()] or [p.strip() for p in text.split('.') if p.strip()]
-    if preview: lines=lines[:1]
-
-    max_slides = 3 if not preview else 1
-    chunk_size = max(1,(len(lines)+max_slides-1)//max_slides)
-    clips=[]
-    try:
-        for i in range(0,len(lines),chunk_size):
-            chunk = " ".join(lines[i:i+chunk_size]).strip()
-            if not chunk: continue
-            slide_path=os.path.join(tmp,f"slide_{i}.png")
-            W,H=(480,320)
-            img=Image.new("RGB",(W,H),(255,182,193))
-            draw=ImageDraw.Draw(img)
 
-            try:
-                font = ImageFont.truetype("DejaVuSans-Bold.ttf",18)
-                title_font = ImageFont.truetype("DejaVuSans-Bold.ttf",24)
-            except:
-                font=ImageFont.load_default()
-                title_font=ImageFont.load_default()
-
-            title_text="You are looking beautiful"
-            tw,th=draw.textsize(title_text,font=title_font)
-            draw.text(((W-tw)/2,10),title_text,fill="darkblue",font=title_font)
-
-            margin=20
-            max_w=W-2*margin
-            words=chunk.split()
-            line=""
-            y=60
-            for w in words:
-                test=(line+" "+w).strip()
-                if draw.textsize(test,font=font)[0]<=max_w: line=test
-                else:
-                    draw.text((margin,y),line,fill="black",font=font)
-                    y+=22
-                    line=w
-            if line: draw.text((margin,y),line,fill="black",font=font)
-            img.save(slide_path)
-
-            audio_path=os.path.join(tmp,f"audio_{i}.mp3")
-            gTTS(text=chunk,lang=tts_lang).save(audio_path)
-
-            duration=max(2,min(len(chunk)//12+3,8))
-            clip=ImageClip(slide_path).set_duration(duration).set_audio(AudioFileClip(audio_path))
-            clips.append(clip)
-
-        if not clips: return None
-        final=concatenate_videoclips(clips,method="compose")
-        out=os.path.join(tmp,"output_video.mp4") if not preview else os.path.join(tmp,"preview_video.mp4")
-        final.write_videofile(out,fps=24,codec="libx264",audio_codec="aac",verbose=False,logger=None)
-        return out
-    except:
+# ===========================
+# Video Generator
+# ===========================
+def create_video_simple(text, tts_lang="en", preview=False):
+    if not text.strip():
         return None
 
-# =========================== #
-# Notes + Flowchart           #
-# =========================== #
+    tmp = tempfile.gettempdir()
+
+    lines = [l.strip() for l in text.split(".") if l.strip()]
+    if preview:
+        lines = lines[:1]  # one slide preview
+
+    clips = []
+
+    for i, chunk in enumerate(lines):
+        slide_path = os.path.join(tmp, f"slide_{i}.png")
+
+        # Create slide
+        img = Image.new("RGB", (480, 320), (255, 182, 193))
+        draw = ImageDraw.Draw(img)
+        font = ImageFont.load_default()
+
+        draw.text((20, 20), "AI Generated Slide", fill="blue", font=font)
+        draw.text((20, 80), chunk[:220], fill="black", font=font)
+
+        img.save(slide_path)
+
+        # Audio
+        audio_path = os.path.join(tmp, f"audio_{i}.mp3")
+        gTTS(text=chunk, lang=tts_lang).save(audio_path)
+
+        # Clip
+        clip = ImageClip(slide_path).set_duration(4)
+        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")
+
+    return output_path
+
+
+# ===========================
+# Notes + Flowchart
+# ===========================
 def generate_notes_flowchart(text):
-    if not text.strip(): return "No text provided.",None
-    notes=call_groq(f"Summarize into concise study notes:\n\n{text}",max_tokens=500)
-    img=Image.new("RGB",(480,320),(144,238,144))
-    draw=ImageDraw.Draw(img)
-    try: font = ImageFont.truetype("DejaVuSans.ttf",16)
-    except: font=ImageFont.load_default()
-
-    y=20
+    notes = call_groq(f"Summarize into clean notes:\n\n{text}")
+    img = Image.new("RGB", (500, 350), (180, 250, 180))
+    draw = ImageDraw.Draw(img)
+    font = ImageFont.load_default()
+
+    y = 20
     for line in notes.split("\n"):
-        if line.strip():
-            draw.text((10,y),line.strip(),fill="black",font=font)
-            y+=18
-            if y>280: break
-    return notes,img
-
-# =========================== #
-# Search 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_links_clickable(topic):
-    if not topic.strip(): return "No topic provided.",""
-    yt_links=[f"[YouTube Result {i+1}]({get_youtube_link(topic)})" for i in range(3)]
-    gg_links=[f"[Google Result {i+1}]({get_google_link(topic)})" for i in range(3)]
-    return "\n".join(yt_links), "\n".join(gg_links)
-
-# =========================== #
-# Gradio Frontend             #
-# =========================== #
-with gr.Blocks() as demo:
-
-    gr.Markdown("## 🌟 Professional AI Assistant Demo (Hackathon)")
-
-    with gr.Tab("Step 1: 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 (demo)",interactive=False)
-        su_btn.click(signup_user,inputs=[su_name,su_email,su_pass,su_contact],outputs=[su_msg,su_code])
-
-    with gr.Tab("Step 2: Verify"):
-        verify_email = gr.Textbox(label="Email")
-        verify_code = gr.Textbox(label="Enter verification code")
-        verify_btn = gr.Button("Verify")
-        verify_msg = gr.Textbox(label="Verification Status",interactive=False)
-        verify_btn.click(verify_user,inputs=[verify_email,verify_code],outputs=[verify_msg])
-
-    with gr.Tab("Step 3: 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",interactive=False)
-        li_success = gr.State(False)
-
-        def login_wrapper(email,password):
-            success,msg=login_user(email,password)
-            return msg,success
-
-        li_btn.click(login_wrapper,inputs=[li_email,li_pass],outputs=[li_msg,li_success])
-
-    with gr.Tab("Step 4: Ask Question"):
-        q_in = gr.Textbox(label="Type your question")
-        ans_out = gr.Textbox(label="Answer",lines=12)
-        mcq_out = gr.Textbox(label="MCQs",lines=12)
-        preview_q_vid = gr.Video(label="Preview Video")
-        full_q_vid = gr.Video(label="Full Video")
-        q_btn = gr.Button("Get Answer")
-
-        def process_question(q):
-            ans = answer_text_query(q)
-            mcq = generate_mcqs_from_text(ans)
-            preview = create_video_simple(ans,preview=True)
-            full = create_video_simple(ans,preview=False)
-            return ans,mcq,preview,full
-
-        q_btn.click(process_question,inputs=[q_in],outputs=[ans_out,mcq_out,preview_q_vid,full_q_vid])
-
-    with gr.Tab("Step 5: Upload Document"):
-        file_in = gr.File(label="Upload page image")
-        sel = gr.CheckboxGroup(["Extracted Text","MCQs","Video","Notes & Flowchart"])
-        lang = gr.Dropdown(label="Video language",choices=["en","ur","hi"],value="en")
-        out_text = gr.Textbox(label="Extracted / Answer Text",lines=12)
-        out_mcq = gr.Textbox(label="MCQs",lines=12)
-        preview_vid = gr.Video(label="Preview Video")
-        full_vid = gr.Video(label="Full Video")
-        notes_out = gr.Textbox(label="Notes",lines=12)
-        flowchart_out = gr.Image(label="Flowchart")
-
-        def process_upload(file,outputs_selected,lang_code):
-            text=""
-            try: text=pytesseract.image_to_string(Image.open(file.name))
-            except: text=""
-
-            mcq_text=""
-            preview_path=None
-            video_path=None
-            notes_text=""
-            flowchart_img=None
-
-            if "MCQs" in outputs_selected and text.strip():
-                mcq_text=generate_mcqs_from_text(text)
-
-            if "Video" in outputs_selected and text.strip():
-                preview_path=create_video_simple(text,tts_lang=lang_code,preview=True)
-                video_path=create_video_simple(text,tts_lang=lang_code,preview=False)
-
-            if "Notes & Flowchart" in outputs_selected and text.strip():
-                notes_text,flowchart_img=generate_notes_flowchart(text)
-
-            return text or "",mcq_text or "",preview_path,video_path,notes_text,flowchart_img
+        draw.text((10, y), line[:60], 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
+
+
+# ===========================
+# UI
+# ===========================
+with gr.Blocks(title="AI Study Wizard") as demo:
+
+    gr.Markdown("# 🎓 AI Study Wizard — Full 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"):
+        v_email = gr.Textbox(label="Email")
+        v_code = gr.Textbox(label="Code")
+        v_btn = gr.Button("Verify")
+        v_msg = gr.Textbox(label="Status")
+        v_btn.click(verify_user, [v_email, v_code], v_msg)
+
+    # 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):
+            try:
+                text = pytesseract.image_to_string(Image.open(f.name))
+            except:
+                text = ""
 
-        btn = gr.Button("Process")
-        btn.click(process_upload,inputs=[file_in,sel,lang],
-                  outputs=[out_text,out_mcq,preview_vid,full_vid,notes_out,flowchart_out])
+            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
 
-    with gr.Tab("Step 6: Links"):
-        topic = gr.Textbox(label="Search topic")
-        yt = gr.Markdown(label="YouTube links")
-        gg = gr.Markdown(label="Google links")
-        sbtn = gr.Button("Get Links")
-        sbtn.click(get_links_clickable,inputs=[topic],outputs=[yt,gg])
+            return text, mcqs, notes, img, prev, full
 
-demo.launch(share=True)
+        btn = gr.Button("Process")
+        btn.click(process_upload, [file, lang],
+                  [out_text, out_mcq, out_notes, out_flow, out_prev, out_full])
+
+    # Search Links
+    with gr.Tab("Search"):
+        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])
+
+demo.launch()