Update app.py

app.py

@@ -1,124 +1,147 @@
 import gradio as gr
 import spaces
 from transformers import pipeline
-import fitz  # PyMuPDF for PDF
+import fitz  # PyMuPDF for PDF reading
 
 # -------------------------------
-# Models
+# Load Models
 # -------------------------------
+# Text AI detector
 text_detector = pipeline("text-classification", model="roberta-base-openai-detector")
+
+# Image classifier (placeholder for "AI detection", but works publicly)
 image_analyzer = pipeline("image-classification", model="microsoft/resnet-50")
+
+# Whisper tiny for audio transcription
 asr_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny", device=-1)
 
 # -------------------------------
-# Functions
+# Oracle Function
 # -------------------------------
 @spaces.GPU
-def oracle_text(user_text: str):
-    if not user_text.strip():
-        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 🤖✨"
-    return f"☀️ **Prophecy:** {score}% Human‑written 🧑‍💻"
+def oracle_prophecy(user_text, user_img, user_audio, user_pdf):
 
-@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}**"
+    prophecy = ""
 
-@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)
-    if label.lower() == "fake":
-        return f"🔊 Transcript: *{transcript}*\n\n🌌 Prophecy: {score}% AI‑generated 🤖"
-    return f"🔊 Transcript: *{transcript}*\n\n☀️ Prophecy: {score}% Human‑written 🧑‍💻"
+    # TEXT check
+    if user_text and user_text.strip():
+        result = text_detector(user_text)
+        label = result[0]['label']
+        score = round(result[0]['score'] * 100, 2)
+        if label.lower() == "fake":
+            prophecy += f"📜 Text: 🌌 **Prophecy:** {score}% AI‑generated 🤖✨\n\n"
+        else:
+            prophecy += f"📜 Text: ☀️ **Prophecy:** {score}% Human‑written 🧑‍💻\n\n"
 
-@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])
-    label = result[0]['label']
-    score = round(result[0]['score'] * 100, 2)
-    if label.lower() == "fake":
-        return f"📑 PDF Prophecy: {score}% AI‑generated 🤖"
-    return f"📑 PDF Prophecy: {score}% Human‑authored 🧑‍💻"
+    # IMAGE check
+    if user_img is not None:
+        result = image_analyzer(user_img)
+        label = result[0]['label']
+        score = round(result[0]['score'] * 100, 2)
+        prophecy += f"🖼️ Image: Oracle’s Sight → {score}% match with **{label}** 🌠\n\n"
+
+    # AUDIO check
+    if user_audio is not None:
+        transcript = asr_pipe(user_audio)["text"]
+        result = text_detector(transcript)
+        label = result[0]['label']
+        score = round(result[0]['score'] * 100, 2)
+        if label.lower() == "fake":
+            prophecy += f"🔊 Voice transcript: *{transcript}* — 🌌 {score}% AI‑generated 🤖\n\n"
+        else:
+            prophecy += f"🔊 Voice transcript: *{transcript}* — ☀️ {score}% Human‑spoken 🧑‍💻\n\n"
+
+    # PDF check
+    if user_pdf is not None:
+        doc = fitz.open(user_pdf)
+        text = "".join([page.get_text() for page in doc])
+        if text.strip():
+            result = text_detector(text[:800])  # only analyze first 800 chars
+            label = result[0]['label']
+            score = round(result[0]['score'] * 100, 2)
+            if label.lower() == "fake":
+                prophecy += f"📑 PDF: 🌌 Prophecy: {score}% AI‑generated 🤖\n\n"
+            else:
+                prophecy += f"📑 PDF: ☀️ Prophecy: {score}% Human‑authored 🧑‍💻\n\n"
+        else:
+            prophecy += "📑 PDF: ⚠️ No readable text found.\n\n"
+
+    if prophecy.strip() == "":
+        prophecy = "⚠️ Please provide text, image, audio, or PDF for the Oracle."
+
+    return f"<div class='result-box'>{prophecy}</div>"
 
 # -------------------------------
-# UI (copied styling from awesome-pet-talk)
+# Gradio UI (Theme like Awesome‑Pet‑Talk)
 # -------------------------------
 with gr.Blocks(css="""
+/* Background Gradient */
 body {
   background: linear-gradient(135deg, #fbc2eb, #a6c1ee);
   font-family: 'Trebuchet MS', sans-serif;
   color: #2c003e;
   text-align: center;
 }
-h1 {
+
+/* Title + subtitle */
+#title {
   font-size: 3em !important;
   color: #ff0080;
   text-shadow: 2px 2px 8px #00000055;
+  margin-top: 20px;
 }
-h2, h3, label {
+#subtitle {
   font-size: 1.4em !important;
   color: #4b0082;
+  margin-bottom: 20px;
 }
+
+/* Bigger inputs */
 textarea, input, .gr-textbox {
   font-size: 1.2em !important;
 }
+
+/* Results Box */
 .result-box {
-  background: rgba(255,255,255,0.9);
+  background: rgba(255,255,255,0.95);
   border-radius: 20px;
   padding: 20px;
-  margin-top: 20px;
+  margin-top: 25px;
   font-size: 1.4em;
   color: #222;
   box-shadow: 0px 4px 15px rgba(0,0,0,0.3);
   text-align: left;
+  white-space: pre-line;
 }
 """) as demo:
 
-    gr.Markdown("# 🔮 Oracle of Truth 🔮")
-    gr.Markdown("### ✨ Multi‑Sight Divination: **Text · Image · Voice · PDF** ✨")
-
-    with gr.Tab("📜 Text Divination"):
-        txt_in = gr.Textbox(lines=8, label="✍️ Enter Text", placeholder="Paste your essay, blog, or snippet…")
-        btn_txt = gr.Button("✨ Reveal Prophecy")
-        txt_out = gr.Markdown(elem_classes="result-box")
-        btn_txt.click(oracle_text, inputs=txt_in, outputs=txt_out)
-
-    with gr.Tab("🖼️ Image Prophecy"):
-        img_in = gr.Image(type="filepath", label="Upload Image 🖼️")
-        btn_img = gr.Button("✨ Reveal Prophecy")
-        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")
-        btn_aud = gr.Button("✨ Reveal Prophecy")
-        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")
-        pdf_out = gr.Markdown(elem_classes="result-box")
-        btn_pdf.click(oracle_pdf, inputs=pdf_in, outputs=pdf_out)
+    # Title
+    gr.HTML("<div id='title'>🔮 Oracle of Truth 🔮</div>")
+    gr.HTML("<div id='subtitle'>✨ Offer your text, image, voice, or PDF — receive the Oracle's prophecy ✨</div>")
+
+    gr.Markdown("""
+👋 Welcome, Seeker!  
+
+Place your offerings before the Oracle:  
+- 📜 Paste *text* (essay, blog, email, story)  
+- 🖼️ Upload an *image*  
+- 🔊 Speak with your *voice*  
+- 📑 Upload a *PDF*  
+
+Then click **Reveal Prophecy ✨** and watch the Oracle’s vision unfold.  
+""")
+
+    with gr.Row():
+        with gr.Column():
+            txt_in = gr.Textbox(lines=6, label="📝 Text Offering")
+            img_in = gr.Image(type="filepath", label="🖼️ Image Offering")
+            aud_in = gr.Audio(sources=["microphone"], type="filepath", label="🎤 Voice Offering")
+            pdf_in = gr.File(file_types=[".pdf"], label="📄 PDF Offering")
+            btn = gr.Button("✨ Reveal Prophecy", variant="primary")
+
+        with gr.Column():
+            result = gr.HTML(label="Oracle's Prophecy")
+
+    btn.click(oracle_prophecy, inputs=[txt_in, img_in, aud_in, pdf_in], outputs=result)
 
 demo.launch()