Create app.py

app.py

@@ -0,0 +1,135 @@
+import os
+import gradio as gr
+import warnings
+
+from huggingface_hub import Repository
+from huggingface_hub import login
+
+warnings.filterwarnings("ignore")
+
+login(token = os.environ['HF_TOKEN'])
+
+from model import AIContentDetector
+
+repo = Repository(
+    local_dir="",
+    repo_type="dataset",
+    clone_from=os.environ['DATASET'],
+    token=True
+)
+repo.git_pull()
+
+# Initialize the detector
+detector = AIContentDetector()
+
+def analyze_image(image):
+    """Wrapper function for image analysis."""
+    if image is None:
+        return "Please upload an image", 0
+    
+    result, confidence = detector.detect_ai_image(image)
+    return result, confidence
+
+def analyze_video(video):
+    """Wrapper function for video analysis."""
+    if video is None:
+        return "Please upload a video", 0
+    
+    result, confidence = detector.detect_ai_video(video)
+    return result, confidence
+
+def analyze_audio(audio):
+    """Wrapper function for audio analysis."""
+    if audio is None:
+        return "Please upload an audio file", 0
+    
+    result, confidence = detector.detect_ai_audio(audio)
+    return result, confidence
+
+# Create the Gradio interface
+with gr.Blocks(title="GuanFu - AI Content Detector", theme=gr.themes.Soft()) as app:
+    gr.Markdown(
+        """
+        # 🕵️ GuanFu - AI Content Detector
+        
+        Upload images, videos, or audio files to detect if they were generated by AI.
+        
+        **Note**: This is a demonstration app. Real AI detection requires specialized models trained on large datasets of AI vs human-generated content.
+        """
+    )
+    
+    with gr.Tabs():
+        # Image Detection Tab
+        with gr.TabItem("🖼️ Image Detection"):
+            gr.Markdown("### Upload an image to check if it's AI-generated")
+            
+            with gr.Row():
+                with gr.Column():
+                    image_input = gr.Image(type="pil", label="Upload Image")
+                    image_button = gr.Button("Analyze Image", variant="primary")
+                
+                with gr.Column():
+                    image_result = gr.Textbox(label="Detection Result", interactive=False)
+                    image_confidence = gr.Number(label="Confidence Score (%)", interactive=False)
+            
+            image_button.click(
+                fn=analyze_image,
+                inputs=image_input,
+                outputs=[image_result, image_confidence]
+            )
+        
+        # Video Detection Tab
+        with gr.TabItem("🎥 Video Detection"):
+            gr.Markdown("### Upload a video to check if it's AI-generated")
+            
+            with gr.Row():
+                with gr.Column():
+                    video_input = gr.Video(label="Upload Video")
+                    video_button = gr.Button("Analyze Video", variant="primary")
+                
+                with gr.Column():
+                    video_result = gr.Textbox(label="Detection Result", interactive=False)
+                    video_confidence = gr.Number(label="Confidence Score (%)", interactive=False)
+            
+            video_button.click(
+                fn=analyze_video,
+                inputs=video_input,
+                outputs=[video_result, video_confidence]
+            )
+        
+        # Audio Detection Tab
+        with gr.TabItem("🎵 Audio Detection"):
+            gr.Markdown("### Upload an audio file to check if it's AI-generated")
+            
+            with gr.Row():
+                with gr.Column():
+                    audio_input = gr.Audio(type="filepath", label="Upload Audio")
+                    audio_button = gr.Button("Analyze Audio", variant="primary")
+                
+                with gr.Column():
+                    audio_result = gr.Textbox(label="Detection Result", interactive=False)
+                    audio_confidence = gr.Number(label="Confidence Score (%)", interactive=False)
+            
+            audio_button.click(
+                fn=analyze_audio,
+                inputs=audio_input,
+                outputs=[audio_result, audio_confidence]
+            )
+    
+    gr.Markdown(
+        """
+        ---
+        ### How it works:
+        - **Images**: Analyzes statistical properties, dimensions, and visual patterns
+        - **Videos**: Examines multiple frames and temporal consistency
+        - **Audio**: Analyzes spectral features and regularity patterns
+        
+        ### Limitations:
+        - This is a demonstration using heuristic methods
+        - Production systems would use specialized AI detection models
+        - Results are estimates and should not be considered definitive
+        """
+    )
+
+if __name__ == "__main__":
+    app.launch(share=True, server_name="0.0.0.0")