Create app.py

app.py

@@ -0,0 +1,272 @@
+# =========================== #
+#       Install Packages      #
+# =========================== #
+import os, tempfile, random
+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           #
+# =========================== #
+users_db = {}
+verification_codes = {}
+
+# =========================== #
+# Signup & Verification       #
+# =========================== #
+def signup_user(username,email,password,contact):
+    if email in users_db:
+        return "Email already registered. Please login.", ""
+    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):
+    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! You can now login."
+    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,"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"
+
+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
+    }
+    try:
+        r = requests.post(url,json=payload,headers=headers,timeout=30)
+        if r.status_code == 200:
+            return r.json().get("choices",[{}])[0].get("message",{}).get("content","")
+        return f"[Model error {r.status_code}] {r.text}"
+    except Exception as e:
+        return f"[API error] {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}"
+    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}"
+    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:
+        return None
+
+# =========================== #
+# 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
+    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
+
+        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])
+
+    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])
+
+demo.launch(share=True)