Update app.py

app.py

@@ -3,10 +3,52 @@ import os
 from dotenv import load_dotenv
 from supplemental import EnhancedAIAgent, ProjectConfig
 from json import JSONDecodeError  # Add this line to import JSONDecodeError
+from transformers import pipeline
+
+class JSONDecodeError(Exception):
+    pass
 
 # Load environment variables
 load_dotenv()
 
+class JSONDecodeError(Exception):
+    pass
+
+def generate_project_config(project_description):
+    generator = pipeline('text-generation', model='gpt2')
+    
+    prompt = f"Generate a project configuration for the following project description: {project_description}\n\n{{"
+    
+    try:
+        response = generator(prompt, max_length=500, num_return_sequences=1)
+        generated_text = response[0]['generated_text']
+        
+        # Extract the JSON part
+        json_start = generated_text.index('{')
+        json_end = generated_text.rindex('}') + 1
+        config_str = generated_text[json_start:json_end]
+        
+        config = json.loads(config_str)
+        return config
+    except JSONDecodeError as e:
+        st.error(f"Error decoding JSON: {str(e)}")
+        return None
+    except Exception as e:
+        st.error(f"An error occurred: {str(e)}")
+        return None
+
+st.title("Project Configuration Generator")
+
+project_description = st.text_area("Enter your project description:")
+
+if st.button("Generate Configuration"):
+    if project_description:
+        config = generate_project_config(project_description)
+        if config:
+            st.json(config)
+    else:
+        st.warning("Please enter a project description.")
+
 # Initialize EnhancedAIAgent
 agent = EnhancedAIAgent(
     name="WebDevMaster",