improve submission

app.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from dotenv import load_dotenv
-from tasks import text, image, audio
+from tasks import text
 
 # Load environment variables
 load_dotenv()
@@ -9,26 +9,19 @@ app = FastAPI(
     title="Frugal AI Challenge API",
     description="API for the Frugal AI Challenge evaluation endpoints"
 )
-from fastapi import FastAPI
-from tasks.text import router as text_router
 
 @app.get("/health")
 async def health_check():
     return {"status": "ok"}
 
-app.include_router(text_router)
-# Include all routers
-# app.include_router(text.router)
-# app.include_router(image.router)
-# app.include_router(audio.router)
+# Include text router
+app.include_router(text.router)
 
 @app.get("/")
 async def root():
     return {
         "message": "Welcome to the Frugal AI Challenge API",
         "endpoints": {
-            "text": "/text - Text classification task"#,
-#            "image": "/image - Image classification task (coming soon)",
-#            "audio": "/audio - Audio classification task (coming soon)"
+            "text": "/text - Text classification task"
         }
-    } 
+    }

tasks/text.py

@@ -143,4 +143,9 @@ async def evaluate_text(request: TextEvaluationRequest):
             
         except Exception as e:
             logger.error(f"Error during evaluation: {str(e)}")
-            stop_tracking()
+            stop_tracking()
+            raise HTTPException(status_code=500, detail=str(e))
+            
+    except Exception as e:
+        logger.error(f"Error in evaluate_text: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))