fix: importing answerformat class in the app module.

app.py

@@ -5,7 +5,7 @@ import requests
 import inspect
 import pandas as pd
 import logging
-from universal_agent import agent
+from universal_agent import AnswerFormat
 # Setup logger
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)

universal_agent.py

@@ -4,7 +4,7 @@ import pickle
 from io import BytesIO
 from pathlib import Path
 from typing import Any
-import os
+# import os
 
 import cv2
 import polars as pl
@@ -21,7 +21,7 @@ from smolagents import (
     VisitWebpageTool,
     WikipediaSearchTool,
 )
-os.environ["OPENAI_API_KEY"] = "sk-dumykey"
+# os.environ["OPENAI_API_KEY"] = "sk-dumykey"
 # import nest_asyncio
 # nest_asyncio.apply()