Update app.py

app.py

@@ -1,19 +1,11 @@
+import streamlit as st
 import os
-import logging
-from flask import Flask, request, jsonify, render_template
 from dotenv import load_dotenv
 from supplemental import EnhancedAIAgent, ProjectConfig
 
 # Load environment variables
 load_dotenv()
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# Initialize Flask app
-app = Flask(__name__)
-
 # Initialize EnhancedAIAgent
 agent = EnhancedAIAgent(
     name="WebDevMaster",
@@ -22,102 +14,79 @@ agent = EnhancedAIAgent(
     model_name="gpt2"
 )
 
-@app.route('/')
-def index():
-    return render_template('index.html')
+st.title("EnhancedAIAgent Web Development Assistant")
+
+st.write("""
+This is a powerful AI-driven web development assistant that can help you with various tasks such as:
+- Generating project configurations
+- Creating project structures
+- Implementing features
+- Reviewing and optimizing code
+- Generating documentation
+- Suggesting tests and refactoring improvements
+""")
 
-@app.route('/generate_project_config', methods=['POST'])
-def generate_project_config():
-    try:
-        project_description = request.json['project_description']
+option = st.selectbox(
+    "What would you like to do?",
+    ("Generate Project Config", "Create Project Structure", "Implement Feature", 
+     "Review Code", "Optimize Code", "Generate Documentation", 
+     "Suggest Tests", "Explain Code", "Suggest Refactoring")
+)
+
+if option == "Generate Project Config":
+    project_description = st.text_area("Enter project description:")
+    if st.button("Generate"):
         config = agent.generate_project_config(project_description)
-        return jsonify(config.__dict__), 200
-    except Exception as e:
-        logger.error(f"Error generating project config: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.json(config.__dict__)
 
-@app.route('/create_project_structure', methods=['POST'])
-def create_project_structure():
-    try:
-        config_dict = request.json
+elif option == "Create Project Structure":
+    config_json = st.text_area("Enter ProjectConfig JSON:")
+    if st.button("Create"):
+        config_dict = eval(config_json)
         config = ProjectConfig(**config_dict)
         structure = agent.create_project_structure(config)
-        return jsonify(structure), 200
-    except Exception as e:
-        logger.error(f"Error creating project structure: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.json(structure)
 
-@app.route('/implement_feature', methods=['POST'])
-def implement_feature():
-    try:
-        feature_description = request.json['feature_description']
-        existing_code = request.json.get('existing_code')
+elif option == "Implement Feature":
+    feature_description = st.text_area("Enter feature description:")
+    existing_code = st.text_area("Enter existing code (optional):")
+    if st.button("Implement"):
         new_code = agent.implement_feature(feature_description, existing_code)
-        return jsonify({"code": new_code}), 200
-    except Exception as e:
-        logger.error(f"Error implementing feature: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.code(new_code)
 
-@app.route('/review_code', methods=['POST'])
-def review_code():
-    try:
-        code = request.json['code']
+elif option == "Review Code":
+    code = st.text_area("Enter code to review:")
+    if st.button("Review"):
         review = agent.review_code(code)
-        return jsonify({"review": review}), 200
-    except Exception as e:
-        logger.error(f"Error reviewing code: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.write(review)
 
-@app.route('/optimize_code', methods=['POST'])
-def optimize_code():
-    try:
-        code = request.json['code']
-        optimization_goal = request.json['optimization_goal']
+elif option == "Optimize Code":
+    code = st.text_area("Enter code to optimize:")
+    optimization_goal = st.text_input("Enter optimization goal:")
+    if st.button("Optimize"):
         optimized_code = agent.optimize_code(code, optimization_goal)
-        return jsonify({"optimized_code": optimized_code}), 200
-    except Exception as e:
-        logger.error(f"Error optimizing code: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.code(optimized_code)
 
-@app.route('/generate_documentation', methods=['POST'])
-def generate_documentation():
-    try:
-        code = request.json['code']
+elif option == "Generate Documentation":
+    code = st.text_area("Enter code to document:")
+    if st.button("Generate"):
         documentation = agent.generate_documentation(code)
-        return jsonify({"documentation": documentation}), 200
-    except Exception as e:
-        logger.error(f"Error generating documentation: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.markdown(documentation)
 
-@app.route('/suggest_tests', methods=['POST'])
-def suggest_tests():
-    try:
-        code = request.json['code']
+elif option == "Suggest Tests":
+    code = st.text_area("Enter code to suggest tests for:")
+    if st.button("Suggest"):
         test_suggestions = agent.suggest_tests(code)
-        return jsonify({"test_suggestions": test_suggestions}), 200
-    except Exception as e:
-        logger.error(f"Error suggesting tests: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.write(test_suggestions)
 
-@app.route('/explain_code', methods=['POST'])
-def explain_code():
-    try:
-        code = request.json['code']
+elif option == "Explain Code":
+    code = st.text_area("Enter code to explain:")
+    if st.button("Explain"):
         explanation = agent.explain_code(code)
-        return jsonify({"explanation": explanation}), 200
-    except Exception as e:
-        logger.error(f"Error explaining code: {str(e)}")
-        return jsonify({"error": str(e)}), 400
+        st.write(explanation)
 
-@app.route('/suggest_refactoring', methods=['POST'])
-def suggest_refactoring():
-    try:
-        code = request.json['code']
+elif option == "Suggest Refactoring":
+    code = st.text_area("Enter code to suggest refactoring:")
+    if st.button("Suggest"):
         refactoring_suggestions = agent.suggest_refactoring(code)
-        return jsonify({"refactoring_suggestions": refactoring_suggestions}), 200
-    except Exception as e:
-        logger.error(f"Error suggesting refactoring: {str(e)}")
-        return jsonify({"error": str(e)}), 400
-
-if __name__ == '__main__':
-    app.run(debug=os.getenv('FLASK_DEBUG', 'False') == 'True', host='0.0.0.0', port=int(os.getenv('PORT', 5000)))
+        st.write(refactoring_suggestions)