diff --git a/src/core/conversation_manager.py b/src/core/conversation_manager.py
index 7b7cc8b..885a7ca 100644
--- a/src/core/conversation_manager.py
+++ b/src/core/conversation_manager.py
@@ -4,8 +4,8 @@ Conversation manager module for handling dialog flow and states
 import json
 import time
 from PyQt6.QtCore import QObject, QThread, QTimer
-
 from src.core.input_listener import InputListener
+from xbrain.core.chat import run
 
 
 class ConversationManager(QObject):
@@ -98,8 +98,9 @@ class ConversationManager(QObject):
         
         # Format display based on action type
         if action["type"] == "mouse":
-            status_text = f"Mouse action: {action['event']} at position: {action['position']}"
+           
             self.text_buffer = ""
+
         elif action["type"] == "keyboard":
             current_time = time.time()
             
diff --git a/src/core/few_shot_agent.py b/src/core/few_shot_agent.py
new file mode 100644
index 0000000..046be82
--- /dev/null
+++ b/src/core/few_shot_agent.py
@@ -0,0 +1,10 @@
+from pdb import run
+from typing import Any
+
+
+class FewShotAgent:
+    def __init__(self):
+        self.messages = []
+
+    def __call__(self, *args: Any, **kwds: Any) -> Any:
+          pass
\ No newline at end of file
diff --git a/src/main.py b/src/main.py
index 03ffb09..c620046 100644
--- a/src/main.py
+++ b/src/main.py
@@ -5,6 +5,7 @@ import sys
 from PyQt6.QtWidgets import QApplication
 
 from src.ui.main_window import MainWindow
+from xbrain.utils.config import Config
 
 
 def main():
@@ -13,6 +14,12 @@ def main():
     
     Creates and runs the AutoMate application
     """
+    config = Config()
+    base_url = "https://api.openai-next.com/v1"
+    api_key = "sk-fb4R0ieuTV2OISKX715e7e4a588447F0A6A0AaE6123d16C7"
+    model = "gpt-4o"
+    config.set_openai_config(base_url=base_url, api_key=api_key, model=model)
+
     app = QApplication(sys.argv)
     window = MainWindow()
     window.show()
diff --git a/src/ui/input_area.py b/src/ui/input_area.py
index d0c4861..2310f16 100644
--- a/src/ui/input_area.py
+++ b/src/ui/input_area.py
@@ -2,8 +2,6 @@
 Input area component for user message entry
 """
 from PyQt6.QtWidgets import (QWidget, QTextEdit, QPushButton, QHBoxLayout, QVBoxLayout)
-from PyQt6.QtCore import Qt, pyqtSignal
-from PyQt6.QtGui import QFont, QColor
 
 
 class InputArea(QWidget):
diff --git a/src/ui/main_window.py b/src/ui/main_window.py
index 08ea3c0..692d53a 100644
--- a/src/ui/main_window.py
+++ b/src/ui/main_window.py
@@ -2,8 +2,6 @@
 Main application window for the AutoMate interface
 """
 from PyQt6.QtWidgets import QMainWindow, QWidget, QHBoxLayout, QVBoxLayout, QApplication
-from PyQt6.QtCore import Qt
-
 from src.ui.chat_area import ChatArea
 from src.ui.input_area import InputArea
 from src.ui.profile_widget import ProfileWidget
diff --git a/src/ui/message_widgets.py b/src/ui/message_widgets.py
index 5b8f75d..feb6879 100644
--- a/src/ui/message_widgets.py
+++ b/src/ui/message_widgets.py
@@ -3,7 +3,7 @@ Message widget components for chat interface
 """
 from PyQt6.QtWidgets import (QWidget, QLabel, QHBoxLayout, QVBoxLayout)
 from PyQt6.QtCore import Qt
-from PyQt6.QtGui import QFont, QColor, QPalette
+from PyQt6.QtGui import QFont
 
 
 class MessageWidget(QWidget):
diff --git a/src/ui/profile_widget.py b/src/ui/profile_widget.py
index 85cca60..500914f 100644
--- a/src/ui/profile_widget.py
+++ b/src/ui/profile_widget.py
@@ -2,7 +2,7 @@
 Profile widget component for displaying intern information
 """
 import os
-from PyQt6.QtWidgets import (QWidget, QLabel, QVBoxLayout, QHBoxLayout)
+from PyQt6.QtWidgets import (QWidget, QLabel, QVBoxLayout)
 from PyQt6.QtCore import Qt
 from PyQt6.QtGui import QFont, QPixmap, QColor, QPainter, QPen