Update app.py

app.py

@@ -1,63 +1,152 @@
 import gradio as gr
 import spaces
 from transformers import pipeline
-import fitz  # PDF
+import fitz  # PyMuPDF for PDF
 
 # -------------------------------
-# Model load
+# Load Models
 # -------------------------------
 text_detector = pipeline("text-classification", model="roberta-base-openai-detector")
+image_analyzer = pipeline("image-classification", model="microsoft/resnet-50")
+asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=-1)
 
+# -------------------------------
+# Oracle Functions
+# -------------------------------
 @spaces.GPU
 def oracle_text(user_text: str):
     if not user_text.strip():
-        return "⚠️ Please enter text."
+        return "⚠️ Please provide some text."
     result = text_detector(user_text)
     label = result[0]['label']
     score = round(result[0]['score'] * 100, 2)
-    if label.lower() == "fake":
-        return f"🌌 **Prophecy:** {score}% AI‑generated 🤖✨"
-    else:
-        return f"☀️ **Prophecy:** {score}% Human‑written 🧑‍💻"
+    return (
+        f"🌌 **Prophecy:** {score}% AI‑generated 🤖✨"
+        if label.lower() == "fake"
+        else f"☀️ **Prophecy:** {score}% Human‑written 🧑‍💻"
+    )
+
+@spaces.GPU
+def oracle_image(user_img):
+    if user_img is None:
+        return "⚠️ Please upload an image!"
+    result = image_analyzer(user_img)
+    label = result[0]['label']
+    score = round(result[0]['score'] * 100, 2)
+    return f"🖼️ Oracle’s Sight: {score}% resemblance to **{label}**.\n*(experimental 🌌)*"
+
+@spaces.GPU
+def oracle_audio(audio_file):
+    if audio_file is None:
+        return "⚠️ Please record your voice."
+    transcript = asr_pipe(audio_file)["text"]
+    result = text_detector(transcript)
+    label = result[0]['label']
+    score = round(result[0]['score'] * 100, 2)
+    verdict = (
+        f"🔊 Transcript: *{transcript}*\n\n🌌 **Prophecy:** {score}% AI‑generated 🤖"
+        if label.lower() == "fake"
+        else f"🔊 Transcript: *{transcript}*\n\n☀️ **Prophecy:** {score}% Human‑written 🧑‍💻"
+    )
+    return verdict
+
+@spaces.GPU
+def oracle_pdf(pdf_file):
+    if pdf_file is None:
+        return "⚠️ Please upload a PDF."
+    doc = fitz.open(pdf_file)
+    text = "".join([page.get_text() for page in doc])
+    if not text.strip():
+        return "⚠️ No text found in PDF."
+    result = text_detector(text[:800])  # limit to 800 chars
+    label = result[0]['label']
+    score = round(result[0]['score'] * 100, 2)
+    return (
+        f"📑 PDF Prophecy: {score}% AI‑generated."
+        if label.lower() == "fake"
+        else f"📑 PDF Prophecy: {score}% Human‑authored."
+    )
 
 # -------------------------------
-# UI
+# UI Styling + Layout
 # -------------------------------
 with gr.Blocks(css="""
 body {
-  background: linear-gradient(135deg, #ff9a9e, #fad0c4, #a1c4fd, #c2e9fb);
-  font-family: 'Trebuchet MS', sans-serif;
+    background: linear-gradient(135deg, #ff9a9e, #fad0c4, #a1c4fd, #c2e9fb);
+    font-family: 'Trebuchet MS', sans-serif;
+    color: #FFFFFF;
+}
+#title {
+    text-align: center;
+    font-size: 3em;
+    color: #FFD700;
+    padding: 20px;
+    text-shadow: 2px 2px 8px #000;
+}
+#subtitle {
+    text-align: center;
+    font-size: 1.4em;
+    color: #00FFFF;
+    margin-bottom: 25px;
 }
-#title { text-align: center; font-size: 2.8em; color: #FFD700; text-shadow: 2px 2px 6px black; }
-#subtitle { text-align: center; font-size: 1.3em; color: #00FFFF; margin-bottom: 20px; }
 .result-box {
-  background: rgba(0,0,0,0.7);
-  border-radius: 15px;
-  padding: 20px;
-  font-size: 1.4em;
-  color: #FFFFFF;
-  box-shadow: 0 0 15px #FFD700;
-  margin-top: 15px;
+    background: rgba(0, 0, 0, 0.7);
+    border-radius: 15px;
+    padding: 20px;
+    font-size: 1.4em;
+    color: #FFFFFF;
+    box-shadow: 0 0 15px #FFD700;
+    margin-top: 15px;
+}
+.gr-textbox textarea {
+    font-size: 1.1em !important;
 }
 """) as demo:
 
     gr.HTML("<div id='title'>🔮 Oracle of Truth 🔮</div>")
-    gr.HTML("<div id='subtitle'>Offer your text and receive a glowing prophecy ✨</div>")
-
-    txt_in = gr.Textbox(lines=8, label="📝 Your Text", placeholder="Paste an essay, blog, or message…")
-    btn = gr.Button("✨ Reveal Prophecy", variant="primary")
-    txt_out = gr.Markdown(elem_classes="result-box")
-
-    btn.click(oracle_text, inputs=txt_in, outputs=txt_out)
-
-    gr.Examples(
-        examples=[
-            ["Once upon a time in a galaxy far away…"],
-            ["Artificial Intelligence is reshaping industries globally 🌍"],
-            ["LOL can't believe it's Monday again 😅"],
-            ["This manuscript evaluates Transformer models for classification."]
-        ],
-        inputs=txt_in,
-    )
+    gr.HTML("<div id='subtitle'>Multi‑Sight Divination ✨ Text · Image · Voice · PDF</div>")
+
+    gr.Markdown("""
+👋 **Welcome, Seeker!**  
+
+The Oracle gazes upon your offering — whether **text**, **image**, **voice**, or **PDF** —  
+and reveals its prophecy: 🧑‍💻 Human‑born ☀️ or 🤖 Machine‑conjured.  
+
+👉 Choose a panel below and witness the verdict in a glowing prophecy box.
+""")
+
+    with gr.Tab("📜 Text Divination"):
+        txt_in = gr.Textbox(lines=8, label="Your Text ✍️", placeholder="Paste essay, story, blog, email...")
+        btn_txt = gr.Button("✨ Reveal Prophecy", variant="primary")
+        txt_out = gr.Markdown(elem_classes="result-box")
+        btn_txt.click(oracle_text, inputs=txt_in, outputs=txt_out)
+
+        gr.Examples(
+            examples=[
+                ["Once upon a time in a galaxy far away…"],
+                ["Artificial Intelligence is revolutionizing healthcare globally 🌍"],
+                ["LOL can't believe it's Monday again 😅"],
+                ["This manuscript provides a rigorous evaluation of Transformer architectures."]
+            ],
+            inputs=txt_in
+        )
+
+    with gr.Tab("🖼️ Image Prophecy"):
+        img_in = gr.Image(type="filepath", label="Upload an Image 🖼️")
+        btn_img = gr.Button("✨ Reveal Prophecy", variant="primary")
+        img_out = gr.Markdown(elem_classes="result-box")
+        btn_img.click(oracle_image, inputs=img_in, outputs=img_out)
+
+    with gr.Tab("🔊 Voice Revelation"):
+        aud_in = gr.Audio(sources=["microphone"], type="filepath", label="🎙️ Speak your truth")
+        btn_aud = gr.Button("✨ Reveal Prophecy", variant="primary")
+        aud_out = gr.Markdown(elem_classes="result-box")
+        btn_aud.click(oracle_audio, inputs=aud_in, outputs=aud_out)
+
+    with gr.Tab("📑 PDF Prophecy"):
+        pdf_in = gr.File(label="📄 Upload PDF", file_types=[".pdf"])
+        btn_pdf = gr.Button("✨ Reveal Prophecy", variant="primary")
+        pdf_out = gr.Markdown(elem_classes="result-box")
+        btn_pdf.click(oracle_pdf, inputs=pdf_in, outputs=pdf_out)
 
 demo.launch()