Update app.py

app.py

@@ -1,22 +1,9 @@
 import gradio as gr
 from transformers import pipeline
-import openai
-import os
-from dotenv import load_dotenv
 
-# Load environment variables from .env file
-load_dotenv()
-
-# Get OpenAI API key
-openai_api_key = os.getenv("OPENAI_API_KEY")
-if not openai_api_key:
-    raise ValueError("❌ OPENAI_API_KEY environment variable not set.")
-
-# Initialize OpenAI client
-client = openai.OpenAI(api_key=openai_api_key)
-
-# Load local model for code classification
+# Load models
 code_analyzer = pipeline("text-classification", model="microsoft/codebert-base")
+nlp_model = pipeline("text2text-generation", model="google/flan-t5-base")
 
 # Code Review Function
 def analyze_code(code):
@@ -25,31 +12,22 @@ def analyze_code(code):
     result = code_analyzer(code)
     return result[0]["label"], "Consider refactoring for better performance", "Medium"
 
-# Metadata Validator (Mock)
+# Metadata Validator (Mock for now)
 def validate_metadata(metadata):
     if not metadata.strip():
         return "No metadata provided.", "", ""
     return "Field", "Unused field detected", "Remove it to improve performance"
 
-# Natural Language Processor using OpenAI GPT-3.5
+# Natural Language Processor (AI-only, no default)
 def process_nlp_query(query):
     if not query.strip():
         return "No query provided."
-    try:
-        response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant specialized in Salesforce Apex and metadata."},
-                {"role": "user", "content": query}
-            ]
-        )
-        return response.choices[0].message.content.strip()
-    except Exception as e:
-        return f"❌ OpenAI API error: {str(e)}"
+    result = nlp_model(query, max_length=60, do_sample=False)
+    return result[0]["generated_text"]
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 AI Code Review & Metadata Validator (GPT-Powered)")
+    gr.Markdown("# 🤖 AI Code Review & Metadata Validator")
 
     with gr.Tab("Code Review"):
         code_input = gr.Textbox(label="Apex / LWC Code", lines=8)