config.py

import os
from dotenv import load_dotenv

load_dotenv()

API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")

MODELS = [
    "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
    # Add other models here if needed
]

STAGES = [
    {"name": "Idea Name", "field": "idea_name", "question": "Do you have a name for this idea?", "example": "Sharks with Laserbeams"},
    {"name": "Idea Overview", "field": "idea_overview", "question": "Tell me about your idea, give a very simple overview as you might explain to younger child.", "example": "My idea is to attach laserbeams to sharks to create the ultimate evil weapon. The laserbeams are powered by a battery that are recharged by the shark's swimming motion. The laserbeams are used to attack enemy submarines."},
    {"name": "Problem Solution", "field": "problem_solution", "question": "Tell me what Customer problem your idea addresses, and how does your idea address it?", "example": "Our customers' secret lairs are often attacked by submarines. If our clients had sharks with laserbeams they could attack and destroy the submarines, saving the secret lair from being discovered."},
    {"name": "Target Customers", "field": "target_customers", "question": "What customers might be interested in this idea? Or is this a internal cost-savings / efficiency idea?", "example": "Evil geniuses, tyrannical dictators, rogue government agencies"},
    {"name": "Market Value", "field": "market_value", "question": "What markets or industries is this idea for?", "example": "Defense Contractors, submarine manufacturers"},
    {"name": "Existing Solutions", "field": "existing_solutions", "question": "What are some similar ideas or products that already exist today?", "example": "Nuclear Attack submarines and giant squids are somewhat similar to Shark with Laserbeams."},
    {"name": "Unique Value", "field": "unique_value", "question": "How is your idea better or different?", "example": "Shark with Laserbeams are cheaper than Nuclear Attack submarines and the lasers are more powerful than squid tentacles."},
    {"name": "Technical Challenges", "field": "technical_challenges", "question": "Are there any difficult technical problems with making this idea happen?", "example": "We would need to design a waterproof housing for the laser and batteries. We might need to experiment with how powerful lasers need to be when underwater in order to melt submarine steel hulls. We need to explore how to capture sharks and install the waterproof harnesses for the lasers."},
    {"name": "Legal Barriers", "field": "legal_barriers", "question": "Any legal barriers?", "example": "We will need to obtain a weapons export license if we want to sell outside the United States."},
    {"name": "Data Dependencies", "field": "data_dependencies", "question": "Any data dependencies?", "example": "We might need to get data on underwater motion vectors for creating the aiming mechanism for the lasers."},
    {"name": "Team Roles", "field": "team_roles", "question": "What kind of people / job roles do you need to make this idea happen?", "example": "We will need marine biologists and fishermen for obtaining sharks. We will need electrical engineers, optical engineers and software engineers to create the laserbeam weapon. We will need project manager to help manage the budget and schedule. And we need a QA tester to ensure the laser is working underwater."},
    {"name": "Timeline", "field": "timeline", "question": "About how long will it take to build the main components of your idea?", "example": "It will take a 2-3weeks to capture a shark. It will take 2-3 weeks to modify the harness and calibrate the laserbeam. I estimate it will take 3 months to train the shark to attack submarines."},
    {"name": "Additional Info", "field": "additional_info", "question": "Anything else you'd want to add to help us understand and value your idea?", "example": "We might consider adding optional racing stripes or bioluminescent patterns ot the sharks as an upgrade with charge the customer for. We could explore a subscription model where we send the customers a new shark each quarter to replace sharks that age out of the program."}
]

DEFAULT_SYSTEM_PROMPT = """
You are Myamoto, an AI assistant designed to guide users through the innovative idea development process. Your task is to help refine and expand their ideas, focusing strictly on one stage at a time. Be encouraging, creative, and analytical in your responses. Remember to:

1. Analyze the user's input carefully and extract relevant information for the current stage only.
2. Provide constructive feedback and suggestions to improve the idea, specific to the current stage.
3. If the user's response is insufficient, ask follow-up questions to gather more details.
4. Do not move to the next stage automatically. The user must explicitly indicate when they want to move on.
5. Maintain a supportive and enthusiastic tone throughout the conversation.
6. Always include a <form_data></form_data> section in your response with the extracted information for the current stage.
7. If the user tries to skip ahead or provide information for future stages, gently redirect them to the current stage.
8. Use analogies, examples, or thought-provoking questions to help users think more deeply about their ideas.
9. When you feel the current stage has been thoroughly explored, explicitly state "stage complete" or "ready to move on" in your response.
10. If you think additional information from a web search might be helpful, suggest it to the user. Ask them to use '@' followed by their search query to perform a web search.

Current stage: {current_stage}
Stage prompt: {stage_prompt}
"""

# Other configuration variables can be added here