Update app.py

app.py

@@ -13,6 +13,8 @@ from functools import partial
 import time
 import csv
 from datetime import datetime
+from fastapi import FastAPI
+from starlette.responses import FileResponse
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -439,15 +441,6 @@ def analyze_text(text: str, mode: str, classifier: TextClassifier) -> tuple:
             overall_result
         )
 
-# Add a function to download the logs
-def download_logs():
-    log_path = "/tmp/prediction_logs.csv"
-    if os.path.exists(log_path):
-        with open(log_path, 'r', encoding='utf-8') as f:
-            content = f.read()
-        return content
-    return "No logs found."
-
 # Initialize the classifier globally
 classifier = TextClassifier()
 
@@ -478,24 +471,31 @@ demo = gr.Interface(
     flagging_mode="never"
 )
 
-# Add admin panel for log access (only visible to space owners)
-with gr.Blocks() as admin_interface:
-    gr.Markdown("## Admin Panel - Data Logs")
-    download_button = gr.Button("Download Logs")
-    log_output = gr.File(label="Prediction Logs")
-    download_button.click(fn=download_logs, outputs=log_output)
-
-# Combine interfaces
-app = gr.TabbedInterface([demo, admin_interface], ["AI Text Detector", "Admin"])
+# Get the FastAPI app from Gradio
+app = demo.app
 
-app.app.add_middleware(
+# Add CORS middleware
+app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],  # For development
     allow_credentials=True,
-    allow_methods=["GET", "POST", "OPTIONS"],  # Explicitly list methods
+    allow_methods=["GET", "POST", "OPTIONS"],
     allow_headers=["*"],
 )
 
+# Add FastAPI endpoint for downloading logs
+@app.get("/download-logs")
+async def download_logs():
+    """Endpoint to download the prediction logs CSV file."""
+    log_path = "/tmp/prediction_logs.csv"
+    if os.path.exists(log_path):
+        return FileResponse(
+            path=log_path, 
+            filename="prediction_logs.csv",
+            media_type="text/csv"
+        )
+    return {"error": "No logs found"}
+
 # Ensure CORS is applied before launching
 if __name__ == "__main__":
     demo.queue()