Update supplemental.py

supplemental.py

@@ -99,8 +99,11 @@ class EnhancedAIAgent:
                 project_files[file_path] = content
         return project_files
 
-    def generate_project_config(self, project_description: str) -> ProjectConfig:
-        prompt = f"""
+    import json
+from json.decoder import JSONDecodeError
+
+def generate_project_config(self, project_description: str) -> Optional[ProjectConfig]:
+    prompt = f"""
 Based on the following project description, generate a ProjectConfig object:
 
 Description: {project_description}
@@ -113,16 +116,75 @@ The ProjectConfig should include:
 
 Respond with a JSON object representing the ProjectConfig.
 """
-        response = self.generate_agent_response(prompt)
-
-        try:
-            config_dict = json.loads(response)
+    response = self.generate_agent_response(prompt)
+
+    try:
+        # Try to find and extract a JSON object from the response
+        json_start = response.find('{')
+        json_end = response.rfind('}') + 1
+        if json_start != -1 and json_end != -1:
+            json_str = response[json_start:json_end]
+            config_dict = json.loads(json_str)
             return ProjectConfig(**config_dict)
-        except JSONDecodeError as e:
-            self.logger.error(f"Error decoding JSON: {str(e)}")
-            self.logger.error(f"Response from model: {response}")
-            return None  # Or handle the error differently
-
+        else:
+            raise ValueError("No JSON object found in the response")
+    except (JSONDecodeError, ValueError) as e:
+        self.logger.error(f"Error parsing JSON from response: {str(e)}")
+        self.logger.error(f"Full response from model: {response}")
+        
+        # Attempt to salvage partial information
+        try:
+            partial_config = self.extract_partial_config(response)
+            if partial_config:
+                self.logger.warning("Extracted partial config from malformed response")
+                return partial_config
+        except Exception as ex:
+            self.logger.error(f"Failed to extract partial config: {str(ex)}")
+        
+        return None
+
+def extract_partial_config(self, response: str) -> Optional[ProjectConfig]:
+    """Attempt to extract partial configuration from a malformed response."""
+    name = self.extract_field(response, "name")
+    description = self.extract_field(response, "description")
+    technologies = self.extract_list(response, "technologies")
+    structure = self.extract_dict(response, "structure")
+    
+    if name and description:
+        return ProjectConfig(
+            name=name,
+            description=description,
+            technologies=technologies or [],
+            structure=structure or {}
+        )
+    return None
+
+def extract_field(self, text: str, field: str) -> Optional[str]:
+    """Extract a simple field value from text."""
+    match = re.search(rf'"{field}"\s*:\s*"([^"]*)"', text)
+    return match.group(1) if match else None
+
+def extract_list(self, text: str, field: str) -> Optional[List[str]]:
+    """Extract a list from text."""
+    match = re.search(rf'"{field}"\s*:\s*\[(.*?)\]', text, re.DOTALL)
+    if match:
+        items = re.findall(r'"([^"]*)"', match.group(1))
+        return items
+    return None
+
+def extract_dict(self, text: str, field: str) -> Optional[Dict[str, List[str]]]:
+    """Extract a dictionary from text."""
+    match = re.search(rf'"{field}"\s*:\s*\{{(.*?)\}}', text, re.DOTALL)
+    if match:
+        dict_str = match.group(1)
+        result = {}
+        for item in re.finditer(r'"([^"]*)"\s*:\s*\[(.*?)\]', dict_str, re.DOTALL):
+            key = item.group(1)
+            values = re.findall(r'"([^"]*)"', item.group(2))
+            result[key] = values
+        return result
+    return None
+    
     def implement_feature(self, feature_description: str, existing_code: Optional[str] = None) -> str:
         prompt = f"""
 Feature to implement: {feature_description}