Update app.py

app.py

@@ -42,6 +42,61 @@ scenarios = [
     }
 ]
 
+# AI-generated scenarios templates
+ai_scenario_templates = [
+    {
+        "title_prefix": "Data Analysis with ",
+        "tools": ["Power BI", "Tableau", "Excel", "Python", "R", "SQL"],
+        "data_types": ["sales", "customer", "financial", "survey", "marketing", "operational"],
+        "problems": ["missing values", "outliers", "inconsistent formats", "duplicates", "aggregation needs"],
+        "level_options": ["Beginner", "Intermediate", "Advanced"]
+    },
+    {
+        "title_prefix": "Visualization in ",
+        "tools": ["Power BI", "Tableau", "Matplotlib", "Seaborn", "D3.js", "Excel"],
+        "data_types": ["time series", "categorical", "geospatial", "hierarchical", "correlation", "comparison"],
+        "problems": ["finding the right chart type", "color selection", "handling scale issues", "showing too many variables", "highlighting key insights"],
+        "level_options": ["Beginner", "Intermediate", "Advanced"]
+    },
+    {
+        "title_prefix": "Query Optimization for ",
+        "tools": ["SQL Server", "PostgreSQL", "MySQL", "BigQuery", "Snowflake", "DynamoDB"],
+        "data_types": ["relational", "document", "key-value", "graph", "time series", "data warehouse"],
+        "problems": ["slow joins", "inefficient filters", "missing indexes", "poor aggregation", "subquery complexity"],
+        "level_options": ["Beginner", "Intermediate", "Advanced"]
+    }
+]
+
+# Generate an AI scenario
+def generate_ai_scenario():
+    template = random.choice(ai_scenario_templates)
+    
+    tool = random.choice(template["tools"])
+    data_type = random.choice(template["data_types"])
+    problem = random.choice(template["problems"])
+    level = random.choice(template["level_options"])
+    
+    title = f"{template['title_prefix']}{tool}"
+    
+    description = f"You're working with {data_type} data and encountering issues with {problem}. " + \
+                 f"How would you instruct an AI assistant to help you solve this problem using {tool}?"
+    
+    ideal_prompt = f"Help me address {problem} in my {data_type} data using {tool}. " + \
+                  f"Please provide a step-by-step approach, including how to identify the issue, " + \
+                  f"resolve it efficiently, and verify the solution is working."
+    
+    # Use the sales data as a generic example
+    sample_data = sample_sales_data.head(10).to_html() if random.random() > 0.5 else f"<pre>{sample_sql_query}</pre>"
+    
+    return {
+        "title": title,
+        "description": description,
+        "level": level,
+        "category": tool,
+        "ideal_prompt": ideal_prompt,
+        "sample_data": sample_data
+    }
+
 # Simulated user progress
 user_progress = {
     "completed_scenarios": 0,
@@ -65,7 +120,11 @@ prompt_tips = [
     "Specify how to handle edge cases like missing values or outliers",
     "Include context about the data structure and your goal",
     "Break complex tasks into numbered steps",
-    "Use clear, concise language rather than vague instructions"
+    "Use clear, concise language rather than vague instructions",
+    "Provide examples when possible to clarify your expectations",
+    "Request explanations when solutions are complex",
+    "Ask for alternatives to compare different approaches",
+    "Define success criteria so the AI knows what a good result looks like"
 ]
 
 # Simulate AI evaluation of prompts
@@ -127,9 +186,17 @@ def evaluate_prompt(user_prompt, scenario):
         "feedback": feedback
     }
 
-# Generate a new scenario
-def get_new_scenario():
-    scenario = random.choice(scenarios)
+# Generate a new scenario (mix of predefined and AI-generated)
+def get_new_scenario(use_ai_generated=None):
+    # 50% chance to get an AI-generated scenario if not specified
+    if use_ai_generated is None:
+        use_ai_generated = random.random() > 0.5
+    
+    if use_ai_generated:
+        scenario = generate_ai_scenario()
+    else:
+        scenario = random.choice(scenarios)
+    
     return (
         scenario["title"],
         scenario["description"],
@@ -138,10 +205,24 @@ def get_new_scenario():
         scenario["sample_data"]
     )
 
+# Specifically get an AI-generated scenario
+def get_ai_scenario():
+    return get_new_scenario(use_ai_generated=True)
+
 # Evaluate user input and provide feedback
 def submit_prompt(prompt, title, description, level, category):
-    # Find the scenario
-    scenario = next((s for s in scenarios if s["title"] == title), scenarios[0])
+    # First try to find in predefined scenarios
+    scenario = next((s for s in scenarios if s["title"] == title), None)
+    
+    # If not found, create a temporary scenario object from inputs
+    if scenario is None:
+        scenario = {
+            "title": title,
+            "description": description,
+            "level": level,
+            "category": category,
+            "ideal_prompt": description  # Use description as a fallback for keywords
+        }
     
     # Evaluate prompt
     evaluation = evaluate_prompt(prompt, scenario)
@@ -191,7 +272,8 @@ def display_leaderboard():
 
 # Display random tip
 def get_random_tip():
-    return random.choice(prompt_tips)
+    tip = random.choice(prompt_tips)
+    return f"**Tip:** {tip}"
 
 # Build Gradio interface
 with gr.Blocks(theme=gr.themes.Soft(), title="PromptMaster for Data Analytics") as app:
@@ -210,9 +292,8 @@ with gr.Blocks(theme=gr.themes.Soft(), title="PromptMaster for Data Analytics")
         gr.Markdown("### Sample Data")
         sample_data_output = gr.HTML()
         
-        # Prompt tip
-        with gr.Accordion("Tip for Effective Prompting", open=False):
-            tip_output = gr.Markdown()
+        # Prompt tip box - NOW VISIBLE BY DEFAULT
+        tip_output = gr.Markdown(label="Prompt Tip")
         
         # User input
         prompt_input = gr.Textbox(label="Your Prompt", lines=5, placeholder="Write your prompt here...")
@@ -221,6 +302,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="PromptMaster for Data Analytics")
         with gr.Row():
             submit_btn = gr.Button("Submit Prompt", variant="primary")
             new_scenario_btn = gr.Button("Get New Scenario")
+            ai_scenario_btn = gr.Button("Generate AI Scenario")
             new_tip_btn = gr.Button("Get Prompt Tip")
         
         # Results
@@ -232,6 +314,11 @@ with gr.Blocks(theme=gr.themes.Soft(), title="PromptMaster for Data Analytics")
             outputs=[title_output, description_output, level_output, category_output, sample_data_output]
         )
         
+        ai_scenario_btn.click(
+            get_ai_scenario,
+            outputs=[title_output, description_output, level_output, category_output, sample_data_output]
+        )
+        
         new_tip_btn.click(
             get_random_tip,
             outputs=[tip_output]