Update ontochat/functions.py

ontochat/functions.py

@@ -6,42 +6,31 @@ import gradio as gr
 
 openai_api_key = None
 
-# Global dictionaries to hold pre-identified prompts and their corresponding long prompts
-preidentified_prompts = {
-    "expand_skills_interests": "My skills are [skills] and my interests are [interests]. Can you help me expand these and ensure they align with my occupation?",
-    
-    "expand_user_goal": "My user goal description is [user goal description]. Can you help me refine this goal, without changing the actions or keywords, and ensure it aligns with my interests and domain?",
-    
-    "expand_actions": "The actions I need to take to achieve my goal are [actions]. Can you help me expand these actions, ensuring they align with my occupation and skills, without changing the goal description or keywords?",
-    
-    "expand_keywords": "The keywords for my user goal and actions are [keywords]. Can you help me refine these keywords, ensuring they are relevant to my goal and actions, without changing the goal or actions themselves?",
-    
-    "expand_current_methods": "The current methods I use to perform my actions are [current methods]. Can you help me elaborate on these methods, ensuring they align with my occupation, skills, and goal, without changing the challenges, new methods, or outcomes?",
-    
-    "expand_challenges": "The challenges I face with my current methods are [challenges]. Can you help me expand on these challenges, ensuring they logically relate to my skills and goal, without changing the current methods, new methods, or outcomes?",
-    
-    "expand_new_methods": "The new methods I will use through this ontology to address my challenges are [new methods]. Can you help me elaborate on these new methods, ensuring they align with my occupation, skills, and goal, without changing the current methods, challenges, or outcomes?",
-    
-    "expand_outcomes": "The outcomes I expect after implementing the new methods are [outcomes]. Can you help me refine these outcomes, ensuring they address the challenges and benefit my goal, without changing the current methods, challenges, or new methods?"
-}
-
 current_preidentified_prompts = []
 current_preidentified = [
-    "Cultural Heritage Preservation",
-    "Accessibility and Inclusion",
-    "Multisensory Interactions",
-    "Digital Rights and Preservation"
+    "My name is [Name], I am [Age] years old, and I work as a [Occupation]. My skills include [skills], and I have a strong interest in [interests]. Could you assist me in refining and expanding my skills and interests, ensuring they are aligned with my occupation?",
+    "My user goal description is [user goal description]. Can you help me refine this goal, without changing the actions or keywords, and ensure it aligns with my interests and domain?",
+    "The actions I need to take to achieve my goal are [actions]. Can you help me expand these actions, ensuring they align with my occupation and skills, without changing the goal description or keywords?",
+    "The keywords for my user goal and actions are [keywords]. Can you help me refine these keywords, ensuring they are relevant to my goal and actions, without changing the goal or actions themselves?",
+    "The current methods I use to perform my actions are [current methods]. Can you help me elaborate on these methods, ensuring they align with my occupation, skills, and goal, without changing the challenges, new methods, or outcomes?",
+    "The challenges I face with my current methods are [challenges]. Can you help me expand on these challenges, ensuring they logically relate to my skills and goal, without changing the current methods, new methods, or outcomes?",
+    "The new methods I will use through this ontology to address my challenges are [new methods]. Can you help me elaborate on these new methods, ensuring they align with my occupation, skills, and goal, without changing the current methods, challenges, or outcomes?",
+    "The outcomes I expect after implementing the new methods are [outcomes]. Can you help me refine these outcomes, ensuring they address the challenges and benefit my goal, without changing the current methods, challenges, or new methods?"
     ]
 
 def load_example(selection): 
     return current_preidentified[selection]
-
-def update_examples():
-    current_samples = current_preidentified_prompts
-    return gr.Dataset(samples=current_samples)
     
 def set_openai_api_key(api_key: str):
     global openai_api_key
+    # Strip any leading or trailing spaces
+    api_key = api_key.strip()
+    
+    # Check if the API key contains any spaces or starts with "sk-"
+    if " " in api_key or not api_key.startswith("sk-"):
+        return "Invalid API key format. Please ensure the key does not contain spaces and is a valid OpenAI API key."
+    
+    # Set the API key if the format is valid
     openai_api_key = api_key
     return "API key has been set! Now you can chat with the chatbot. Enjoy :)"
 
@@ -49,9 +38,6 @@ def check_api_key():
     if openai_api_key is None:
         raise ValueError("OpenAI API key is not set. Please set it using the 'Set API Key' button.")
 
-def get_preidentified_prompts():
-    return list(preidentified_prompts.keys())
-
 def user_story_generator(message, history):
     instructions = [
         {
@@ -97,33 +83,8 @@ def user_story_generator(message, history):
         "content": message
     })
     bot_message = chat_completion(openai_api_key, instructions + messages)
-    history.append([message, bot_message])
-    questions = generate_elicitation_questions(history)  
-    return history, "", questions
-
-def generate_elicitation_questions(history):
-    instructions = [{
-        "role": "system",
-        "content": f"In the dictionary {preidentified_prompts}, the key expand_skills_interests corresponds to section 1, '1. Persona'. The keys expand_user_goal, expand_actions, and expand_keywords correspond to section 2, '2. Goal'. The keys expand_current_methods, expand_challenges, expand_new_methods, and expand_outcomes correspond to section 3, '3. Scenario'. All keys in this dictionary are related to section 4, '4. Create User Story'."
-    }]
-    messages = build_messages(history)
-    messages.append({
-        "role": "user",
-        "content": "Identify which section ('1. Persona', '2. Goal', '3. Scenario', '4. Create User Story') your most current answer or elicitation question belongs to."
-                    "If your answer or elicitation question relates to [name], [age], [occupation], [skills], or [interests], it belongs to section 1. If it's about [user goal description], [actions], or [keywords], it belongs to section 2. If it addresses [current methods], [challenges], [new methods], or [outcomes], it belongs to section 3. If sections 1 through 3 are already completed or when the final user story is generated, it belongs to section 4. Once it belongs to section 4, it will always remain in section 4, regardless of what happens afterward."
-                    "Then return all the keys associated with that section. List each key name on a separate line, starting with the first key name and providing only the key names."
-    })
-    bot_message = chat_completion(openai_api_key, instructions + messages)
-    keys = bot_message.split('\n')
-    global current_preidentified_prompts
-    current_preidentified_prompts = [[key] for key in keys if key in preidentified_prompts]
-    global current_preidentified  
-    current_preidentified=[]
-    for sublist in current_preidentified_prompts:
-        key = sublist[0]
-        if key in preidentified_prompts:
-            current_preidentified.append(preidentified_prompts[key])
-    return [[key] for key in keys if key in preidentified_prompts]
+    history.append([message, bot_message])  
+    return history, ""
 
 def cq_generator(message, history):
     check_api_key()