Update app.py

app.py

@@ -213,47 +213,128 @@ def format_audio_results(audio_file):
     res = analyze_audio(audio_file)
     return f"### Audio Detection\nAI Probability: {res['ai_probability']:.4f}\nConfidence: {res['confidence']}\nExplanation: {res['explanation']}"
 
+# with gr.Blocks() as app:
+#     home = gr.Column(visible=True)
+#     with home:
+#         gr.Markdown("## AI Multi-Modal Detector")
+#         with gr.Row():
+#             t_btn = gr.Button("Text")
+#             i_btn = gr.Button("Image")
+#             v_btn = gr.Button("Video")
+#             a_btn = gr.Button("Audio")
+
+#     text_page = gr.Column(visible=False)
+#     with text_page:
+#         inp = gr.Textbox(lines=5, placeholder="Paste text...", label="Text")
+#         out = gr.Markdown()
+#         gr.Button("Analyze").click(format_text_results, inputs=inp, outputs=out)
+#         gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,text_page])
+
+#     image_page = gr.Column(visible=False)
+#     with image_page:
+#         inp = gr.Image(type="pil")
+#         out = gr.Markdown()
+#         gr.Button("Analyze").click(format_image_results, inputs=inp, outputs=out)
+#         gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,image_page])
+
+#     video_page = gr.Column(visible=False)
+#     with video_page:
+#         inp = gr.Video()
+#         out = gr.Markdown()
+#         gr.Button("Analyze").click(format_video_results, inputs=inp, outputs=out)
+#         gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,video_page])
+
+#     audio_page = gr.Column(visible=False)
+#     with audio_page:
+#         inp = gr.Audio(type="filepath")
+#         out = gr.Markdown()
+#         gr.Button("Analyze").click(format_audio_results, inputs=inp, outputs=out)
+#         gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,audio_page])
+
+#     t_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,text_page])
+#     i_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,image_page])
+#     v_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,video_page])
+#     a_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,audio_page])
+
+# app.launch(share=True)
 with gr.Blocks() as app:
-    home = gr.Column(visible=True)
-    with home:
-        gr.Markdown("## AI Multi-Modal Detector")
+    # Home Page
+    home_page = gr.Column(visible=True)
+    with home_page:
+        gr.Markdown("## 🏠 AI Detection Tool")
+        gr.Markdown("Select an option below to continue:")
         with gr.Row():
-            t_btn = gr.Button("Text")
-            i_btn = gr.Button("Image")
-            v_btn = gr.Button("Video")
-            a_btn = gr.Button("Audio")
+            text_page_btn = gr.Button("🧠 Text Detection")
+            image_page_btn = gr.Button("🖼️ Image Detection")
+            video_page_btn = gr.Button("🎬 Video Detection")  # Add on home page
+            audio_page_btn = gr.Button("🎵 Audio Detection")  # Add this to home page
 
+    # Text Page
     text_page = gr.Column(visible=False)
     with text_page:
-        inp = gr.Textbox(lines=5, placeholder="Paste text...", label="Text")
-        out = gr.Markdown()
-        gr.Button("Analyze").click(format_text_results, inputs=inp, outputs=out)
-        gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,text_page])
+        gr.Markdown("## 🧠 Text Detection")
+        text_input = gr.Textbox(lines=5, placeholder="Paste your text here...", label="Input Text")
+        text_output = gr.Markdown("⚡ Result will appear here after submission...", label="Result")
+        analyze_text_btn = gr.Button("Analyze Text")
+        back_btn_text = gr.Button("⬅️ Back")
 
+    # Image Page
     image_page = gr.Column(visible=False)
     with image_page:
-        inp = gr.Image(type="pil")
-        out = gr.Markdown()
-        gr.Button("Analyze").click(format_image_results, inputs=inp, outputs=out)
-        gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,image_page])
-
+        gr.Markdown("## 🖼️ Image Detection")
+        image_input = gr.Image(type="pil", label="Upload Image")
+        image_output = gr.Markdown("⚡ Result will appear here after image upload...", label="Result")
+        analyze_image_btn = gr.Button("Analyze Image")
+        back_btn_image = gr.Button("⬅️ Back")
+    # Video page
     video_page = gr.Column(visible=False)
     with video_page:
-        inp = gr.Video()
-        out = gr.Markdown()
-        gr.Button("Analyze").click(format_video_results, inputs=inp, outputs=out)
-        gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,video_page])
-
+        gr.Markdown("## 🎬 Video Detection")
+        video_input = gr.Video(label="Upload Video")  # Corrected
+        video_output = gr.Markdown("⚡ Result will appear here after video upload...", label="Result")
+        analyze_video_btn = gr.Button("Analyze Video")
+        back_btn_video = gr.Button("⬅️ Back")
     audio_page = gr.Column(visible=False)
     with audio_page:
-        inp = gr.Audio(type="filepath")
-        out = gr.Markdown()
-        gr.Button("Analyze").click(format_audio_results, inputs=inp, outputs=out)
-        gr.Button("Back").click(lambda: (gr.update(visible=True), gr.update(visible=False)), outputs=[home,audio_page])
+        gr.Markdown("## 🎵 Audio Detection")
+        audio_input = gr.Audio(label="Upload Audio", type="filepath")  # Use type="filepath" to get local path
+        audio_output = gr.Markdown("⚡ Result will appear here after audio upload...", label="Result")
+        analyze_audio_btn = gr.Button("Analyze Audio")
+        back_btn_audio = gr.Button("⬅️ Back")
+
+    def show_video_page():
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+    def show_audio_page():
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+
+    audio_page_btn.click(show_audio_page, outputs=[home_page, text_page, image_page, video_page, audio_page])
+
+    # Back button returns to home
+    # Navigation functions
+    def show_text_page():
+        return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+    def show_image_page():
+        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+    def show_home():
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+
+    # Bind navigation buttons
+    text_page_btn.click(show_text_page, outputs=[home_page, text_page, image_page])
+    image_page_btn.click(show_image_page, outputs=[home_page, text_page, image_page])
+    back_btn_text.click(show_home, outputs=[home_page, text_page, image_page])
+    back_btn_image.click(show_home, outputs=[home_page, text_page, image_page])
+    video_page_btn.click(show_video_page, outputs=[home_page, text_page, image_page, video_page])
+    back_btn_video.click(lambda: (gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)),
+                         outputs=[home_page, text_page, image_page, video_page])
+    back_btn_audio.click(lambda: (
+        gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    ), outputs=[home_page, text_page, image_page, video_page, audio_page])
+
 
-    t_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,text_page])
-    i_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,image_page])
-    v_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,video_page])
-    a_btn.click(lambda: (gr.update(visible=False), gr.update(visible=True)), outputs=[home,audio_page])
+    # Bind analysis buttons
+    analyze_text_btn.click(format_text_results, inputs=text_input, outputs=text_output)
+    analyze_image_btn.click(format_image_results, inputs=image_input, outputs=image_output)
+    analyze_video_btn.click(format_video_results, inputs=video_input, outputs=video_output)
+    analyze_audio_btn.click(format_audio_results, inputs=audio_input, outputs=audio_output)
 
-app.launch(share=True)
+app.launch(share=True, debug=True)