Refactor: Move configurations to JSON files and update to GPT-4o

app.py

@@ -8,6 +8,7 @@ from chainlit.prompt import Prompt, PromptMessage  # importing prompt tools
 from chainlit.playground.providers import ChatOpenAI  # importing ChatOpenAI tools
 from dotenv import load_dotenv
 from prompt_manager import PromptManager
+from config_manager import ConfigManager
 from logger_config import logger
 
 # Load environment variables
@@ -17,7 +18,8 @@ logger.info("Environment variables loaded")
 # Initialize OpenAI client
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-# Initialize PromptManager
+# Initialize managers
+config = ConfigManager()
 prompt_manager = PromptManager()
 
 @cl.on_chat_start
@@ -50,12 +52,19 @@ async def main(message: cl.Message):
     ]
     logger.debug("Created messages for API call")
     
+    # Get model configuration
+    model_config = config.get_model_config()
+    logger.debug("Retrieved model configuration")
+    
     # Send message to OpenAI
     response = client.chat.completions.create(
-        model="gpt-3.5-turbo",
+        model=model_config["name"],
         messages=messages,
-        temperature=0.7,
-        max_tokens=1000
+        temperature=model_config["temperature"],
+        max_tokens=model_config["max_tokens"],
+        top_p=model_config["top_p"],
+        frequency_penalty=model_config["frequency_penalty"],
+        presence_penalty=model_config["presence_penalty"]
     )
     logger.info("Received response from OpenAI")
     

aspects.json

@@ -0,0 +1,72 @@
+{
+    "default": {
+        "description": "General purpose AI assistant with balanced capabilities",
+        "examples": [
+            "What can you help me with?",
+            "How do you work?",
+            "What are your capabilities?"
+        ],
+        "system_template": "You are a helpful AI assistant with expertise in various domains.\nYour goal is to provide clear, accurate, and helpful responses while maintaining a friendly tone.\nAdapt your communication style based on the user's needs and the complexity of the topic.",
+        "user_template": "{input}\nThink through your response step by step."
+    },
+    "Concept Simplification": {
+        "description": "Break down complex topics into simple, understandable terms",
+        "examples": [
+            "Explain quantum computing in simple terms",
+            "How does blockchain work?",
+            "What is machine learning?"
+        ],
+        "system_template": "You are an expert at breaking down complex concepts into simple, understandable terms.\nYour goal is to make difficult topics accessible while maintaining accuracy.\nAlways use clear examples and analogies to illustrate points.",
+        "user_template": "Please help me understand this concept: {input}\n- What are the key components?\n- Can you provide a simple analogy?\n- What are common misconceptions?"
+    },
+    "Summarization": {
+        "description": "Extract and present key information concisely",
+        "examples": [
+            "Summarize the key points of climate change",
+            "Give me a brief overview of the internet's history",
+            "What are the main ideas in this article?"
+        ],
+        "system_template": "You are a skilled summarizer who can extract key information and present it concisely.\nFocus on the most important points while maintaining context.\nStructure your summaries in a clear, logical manner.",
+        "user_template": "Please summarize this information: {input}\n- What are the main points?\n- What's the key takeaway?\n- What context is important to retain?"
+    },
+    "Creativity": {
+        "description": "Generate innovative ideas and unique perspectives",
+        "examples": [
+            "Generate ideas for a sustainable city",
+            "How could we improve remote work?",
+            "What are some creative solutions to reduce plastic waste?"
+        ],
+        "system_template": "You are a creative thinker who can generate innovative ideas and solutions.\nThink outside the box while staying relevant to the topic.\nEncourage creative exploration and unique perspectives.",
+        "user_template": "Let's explore this creatively: {input}\n- What are some unique perspectives?\n- How can we approach this differently?\n- What innovative solutions can we consider?"
+    },
+    "Narrative Structure": {
+        "description": "Organize information into compelling stories",
+        "examples": [
+            "Help me structure a story about time travel",
+            "Organize the history of AI as a narrative",
+            "How can I make this presentation more engaging?"
+        ],
+        "system_template": "You are an expert in storytelling and narrative organization.\nHelp structure information in a compelling, story-like format.\nFocus on flow, progression, and engagement in your responses.",
+        "user_template": "Help me structure this as a narrative: {input}\n- What's the main story arc?\n- How can we make it more engaging?\n- What elements would enhance the flow?"
+    },
+    "Arithmetic Reasoning": {
+        "description": "Solve mathematical problems step by step",
+        "examples": [
+            "Calculate compound interest on $1000 at 5% for 3 years",
+            "If a train travels at 60 mph for 2.5 hours, how far does it go?",
+            "What's the probability of getting three heads in a row?"
+        ],
+        "system_template": "You are a mathematical reasoning expert who can solve problems step by step.\nBreak down complex calculations into manageable parts.\nExplain your mathematical thinking clearly and thoroughly.",
+        "user_template": "Let's solve this step by step: {input}\n- What's the first step?\n- What formulas or methods should we use?\n- How can we verify the answer?"
+    },
+    "Conversational Tone": {
+        "description": "Engage in natural, friendly discussions",
+        "examples": [
+            "Tell me about your favorite book",
+            "What's your opinion on artificial intelligence?",
+            "How do you feel about remote work?"
+        ],
+        "system_template": "You are a friendly and engaging conversational partner.\nMaintain a natural, warm tone while being informative.\nAdapt your communication style to match the user's level of expertise.",
+        "user_template": "Let's discuss this: {input}\n- What's your perspective?\n- Can you explain this in simple terms?\n- How does this relate to everyday experience?"
+    }
+} 

chainlit.md

@@ -1,7 +1,7 @@
 # 🤖 Welcome to the Chainlit GPT Assistant!
 
 Hi there! 👋  
-This chatbot is powered by **OpenAI GPT-4 Turbo** and built using **Chainlit** to provide real-time, interactive conversations with streaming responses.
+This chatbot is powered by **OpenAI GPT-4o** and built using **Chainlit** to provide real-time, interactive conversations with streaming responses.
 
 ## 🎯 Conversation Aspects
 
@@ -22,7 +22,7 @@ This chatbot offers multiple conversation aspects to tailor the interaction to y
 
 ## ⚙️ Technical Details
 
-- 🧠 **LLM:** OpenAI GPT-4 Turbo
+- 🧠 **LLM:** OpenAI GPT-4o
 - 🛠️ **Framework:** Chainlit
 - 📝 **Features:** Comprehensive logging system, modular codebase
 - 🌐 **Deployed on:** Hugging Face Spaces

config.json

@@ -0,0 +1,11 @@
+{
+    "model": {
+        "name": "gpt-4o",
+        "temperature": 0.7,
+        "max_tokens": 4000,
+        "top_p": 1.0,
+        "frequency_penalty": 0.0,
+        "presence_penalty": 0.0
+    },
+    "default_aspect": "default"
+} 

config_manager.py

@@ -0,0 +1,45 @@
+import json
+from typing import Dict, Any
+from logger_config import logger
+
+class ConfigManager:
+    _instance = None
+    _config = None
+
+    def __new__(cls):
+        if cls._instance is None:
+            cls._instance = super(ConfigManager, cls).__new__(cls)
+        return cls._instance
+
+    def __init__(self):
+        if self._config is None:
+            self._load_config()
+
+    def _load_config(self) -> None:
+        """Load configuration from JSON file."""
+        try:
+            with open('config.json', 'r') as f:
+                self._config = json.load(f)
+            logger.info("Configuration loaded successfully")
+        except Exception as e:
+            logger.error(f"Error loading configuration: {e}")
+            # Set default configuration
+            self._config = {
+                "model": {
+                    "name": "gpt-4",
+                    "temperature": 0.7,
+                    "max_tokens": 2000,
+                    "top_p": 1.0,
+                    "frequency_penalty": 0.0,
+                    "presence_penalty": 0.0
+                },
+                "default_aspect": "default"
+            }
+
+    def get_model_config(self) -> Dict[str, Any]:
+        """Get model configuration settings."""
+        return self._config.get("model", {})
+
+    def get_default_aspect(self) -> str:
+        """Get default aspect name."""
+        return self._config.get("default_aspect", "default") 

models.py

@@ -0,0 +1,9 @@
+from dataclasses import dataclass
+from typing import List
+
+@dataclass
+class AspectInfo:
+    description: str
+    examples: List[str]
+    system_template: str
+    user_template: str 

prompt_manager.py

@@ -1,164 +1,44 @@
-from dataclasses import dataclass
+import json
 from typing import Dict, List, Optional, Tuple
 from logger_config import logger
-
-# Aspect descriptions and examples
-ASPECT_INFO = {
-    "Concept Simplification": {
-        "description": "Break down complex topics into simple, understandable terms",
-        "examples": [
-            "Explain quantum computing in simple terms",
-            "How does blockchain work?",
-            "What is machine learning?"
-        ]
-    },
-    "Summarization": {
-        "description": "Extract and present key information concisely",
-        "examples": [
-            "Summarize the key points of climate change",
-            "Give me a brief overview of the internet's history",
-            "What are the main ideas in this article?"
-        ]
-    },
-    "Creativity": {
-        "description": "Generate innovative ideas and unique perspectives",
-        "examples": [
-            "Generate ideas for a sustainable city",
-            "How could we improve remote work?",
-            "What are some creative solutions to reduce plastic waste?"
-        ]
-    },
-    "Narrative Structure": {
-        "description": "Organize information into compelling stories",
-        "examples": [
-            "Help me structure a story about time travel",
-            "Organize the history of AI as a narrative",
-            "How can I make this presentation more engaging?"
-        ]
-    },
-    "Arithmetic Reasoning": {
-        "description": "Solve mathematical problems step by step",
-        "examples": [
-            "Calculate compound interest on $1000 at 5% for 3 years",
-            "If a train travels at 60 mph for 2.5 hours, how far does it go?",
-            "What's the probability of getting three heads in a row?"
-        ]
-    },
-    "Conversational Tone": {
-        "description": "Engage in natural, friendly discussions",
-        "examples": [
-            "Tell me about your favorite book",
-            "What's your opinion on artificial intelligence?",
-            "How do you feel about remote work?"
-        ]
-    }
-}
-
-@dataclass
-class AspectInfo:
-    description: str
-    examples: List[str]
-    system_template: str
-    user_template: str
+from models import AspectInfo
+from config_manager import ConfigManager
 
 class PromptManager:
     def __init__(self):
         logger.info("Initializing PromptManager")
-        # Default template for when no aspect is selected
-        self.default_aspect = AspectInfo(
-            description="General purpose AI assistant with balanced capabilities",
-            examples=[
-                "What can you help me with?",
-                "How do you work?",
-                "What are your capabilities?"
-            ],
-            system_template="""You are a helpful AI assistant with expertise in various domains.
-Your goal is to provide clear, accurate, and helpful responses while maintaining a friendly tone.
-Adapt your communication style based on the user's needs and the complexity of the topic.""",
-            user_template="""{input}
-Think through your response step by step."""
-        )
-        logger.debug("Default aspect template created")
-
-        self.aspects = {
-            "Concept Simplification": AspectInfo(
-                description=ASPECT_INFO["Concept Simplification"]["description"],
-                examples=ASPECT_INFO["Concept Simplification"]["examples"],
-                system_template="""You are an expert at breaking down complex concepts into simple, understandable terms.
-Your goal is to make difficult topics accessible while maintaining accuracy.
-Always use clear examples and analogies to illustrate points.""",
-                user_template="""Please help me understand this concept: {input}
-- What are the key components?
-- Can you provide a simple analogy?
-- What are common misconceptions?"""
-            ),
-            "Summarization": AspectInfo(
-                description=ASPECT_INFO["Summarization"]["description"],
-                examples=ASPECT_INFO["Summarization"]["examples"],
-                system_template="""You are a skilled summarizer who can extract key information and present it concisely.
-Focus on the most important points while maintaining context.
-Structure your summaries in a clear, logical manner.""",
-                user_template="""Please summarize this information: {input}
-- What are the main points?
-- What's the key takeaway?
-- What context is important to retain?"""
-            ),
-            "Creativity": AspectInfo(
-                description=ASPECT_INFO["Creativity"]["description"],
-                examples=ASPECT_INFO["Creativity"]["examples"],
-                system_template="""You are a creative thinker who can generate innovative ideas and solutions.
-Think outside the box while staying relevant to the topic.
-Encourage creative exploration and unique perspectives.""",
-                user_template="""Let's explore this creatively: {input}
-- What are some unique perspectives?
-- How can we approach this differently?
-- What innovative solutions can we consider?"""
-            ),
-            "Narrative Structure": AspectInfo(
-                description=ASPECT_INFO["Narrative Structure"]["description"],
-                examples=ASPECT_INFO["Narrative Structure"]["examples"],
-                system_template="""You are an expert in storytelling and narrative organization.
-Help structure information in a compelling, story-like format.
-Focus on flow, progression, and engagement in your responses.""",
-                user_template="""Help me structure this as a narrative: {input}
-- What's the main story arc?
-- How can we make it more engaging?
-- What elements would enhance the flow?"""
-            ),
-            "Arithmetic Reasoning": AspectInfo(
-                description=ASPECT_INFO["Arithmetic Reasoning"]["description"],
-                examples=ASPECT_INFO["Arithmetic Reasoning"]["examples"],
-                system_template="""You are a mathematical reasoning expert who can solve problems step by step.
-Break down complex calculations into manageable parts.
-Explain your mathematical thinking clearly and thoroughly.""",
-                user_template="""Let's solve this step by step: {input}
-- What's the first step?
-- What formulas or methods should we use?
-- How can we verify the answer?"""
-            ),
-            "Conversational Tone": AspectInfo(
-                description=ASPECT_INFO["Conversational Tone"]["description"],
-                examples=ASPECT_INFO["Conversational Tone"]["examples"],
-                system_template="""You are a friendly and engaging conversational partner.
-Maintain a natural, warm tone while being informative.
-Adapt your communication style to match the user's level of expertise.""",
-                user_template="""Let's discuss this: {input}
-- What's your perspective?
-- Can you explain this in simple terms?
-- How does this relate to everyday experience?"""
-            )
-        }
+        self.config = ConfigManager()
+        self.aspects = self._load_aspects()
         logger.info(f"Initialized {len(self.aspects)} aspects")
 
+    def _load_aspects(self) -> Dict[str, AspectInfo]:
+        """Load aspects from JSON file."""
+        try:
+            with open('aspects.json', 'r') as f:
+                aspects_data = json.load(f)
+            
+            aspects = {}
+            for name, data in aspects_data.items():
+                aspects[name] = AspectInfo(
+                    description=data['description'],
+                    examples=data['examples'],
+                    system_template=data['system_template'],
+                    user_template=data['user_template']
+                )
+            return aspects
+        except Exception as e:
+            logger.error(f"Error loading aspects: {e}")
+            return {}
+
     def get_aspect_names(self) -> List[str]:
         """Get list of available aspect names."""
         logger.debug("Getting aspect names")
-        return list(self.aspects.keys())
+        return [name for name in self.aspects.keys() if name != 'default']
 
     def get_aspect_info(self, aspect_name: str) -> AspectInfo:
         """Get information for a specific aspect."""
         logger.debug(f"Getting aspect info for: {aspect_name}")
-        return self.aspects.get(aspect_name, self.default_aspect)
+        return self.aspects.get(aspect_name, self.aspects[self.config.get_default_aspect()])
 
     def get_action_description(self, aspect_name: str) -> str:
         """Get formatted description for action button."""
@@ -183,8 +63,8 @@ Try asking something like:
         logger.debug(f"Getting templates for aspect: {aspect_name}")
         if aspect_name is None:
             logger.info("No aspect selected, using default template")
-            info = self.default_aspect
+            info = self.aspects[self.config.get_default_aspect()]
         else:
             logger.info(f"Using templates for aspect: {aspect_name}")
-            info = self.aspects.get(aspect_name, self.default_aspect)
+            info = self.aspects.get(aspect_name, self.aspects[self.config.get_default_aspect()])
         return info.system_template, info.user_template