Update app.py

app.py

@@ -22,6 +22,15 @@ from agent import (
 from utils import parse_action, parse_file_content, read_python_module_structure
 from datetime import datetime
 
+# Ensure prompts are defined or imported correctly
+# If prompts is a module, import it. If it's a dictionary, define it here.
+# Example:
+prompts = {
+    "WEB_DEV": "Your web development prompt here...",
+    "AI_SYSTEM_PROMPT": "Your AI system prompt here...",
+    "PYTHON_CODE_DEV": "Your Python code development prompt here..."
+}
+
 now = datetime.now()
 date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
 
@@ -230,14 +239,14 @@ def generate(
         prompt, history, agent_name=agents[0], sys_prompt="", temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.7,
 ):
     seed = random.randint(1, 1111111111111111)
-    agent = prompts.WEB_DEV
+    agent = prompts["WEB_DEV"]
 
     if agent_name == "WEB_DEV":
-        agent = prompts.WEB_DEV
+        agent = prompts["WEB_DEV"]
     elif agent_name == "AI_SYSTEM_PROMPT":
-        agent = prompts.AI_SYSTEM_PROMPT
+        agent = prompts["AI_SYSTEM_PROMPT"]
     elif agent_name == "PYTHON_CODE_DEV":
-        agent = prompts.PYTHON_CODE_DEV
+        agent = prompts["PYTHON_CODE_DEV"]
     
     system_prompt = agent
     temperature = float(temperature)