Update src/App.tsx

src/App.tsx

@@ -630,7 +630,6 @@ function PayloadsTab() {
 function BuilderTab() {
   const [prompt, setPrompt] = useState('');
   const [mode, setMode] = useState<'payload' | 'packets' | 'manual' | 'dragdrop'>('dragdrop');
-  const [aiModel, setAiModel] = useState<'2.5' | '3.0'>('2.5');
   const [loading, setLoading] = useState(false);
   const [result, setResult] = useState('');
   const [errorMsg, setErrorMsg] = useState('');
@@ -687,7 +686,7 @@ function BuilderTab() {
     setErrorMsg('');
     try {
       let endpoint = '/api/ai/generate-payload';
-      let body: any = { prompt, model: aiModel };
+      let body: any = { prompt };
       
       if (mode === 'packets') endpoint = '/api/ai/generate-packets';
 
@@ -712,7 +711,7 @@ function BuilderTab() {
       const res = await fetch('/api/ai/analyze-reports', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({ model: aiModel })
+        body: JSON.stringify({})
       });
       const data = await res.json();
       if (data.error) setErrorMsg(data.error);
@@ -791,20 +790,6 @@ function BuilderTab() {
       <div className="grid grid-cols-1 lg:grid-cols-2 gap-6">
         <Card title="AI Input / Configuration" icon={<Cpu size={18} />}>
           <div className="flex flex-col gap-4">
-            <div className="flex bg-black/20 rounded-lg p-1">
-              <button 
-                onClick={() => setAiModel('2.5')}
-                className={cn("flex-1 py-1.5 text-sm rounded-md transition-all font-medium", aiModel === '2.5' ? "bg-slate-700 text-white" : "text-slate-400 hover:text-slate-200")}
-              >
-                Gemini 2.5
-              </button>
-              <button 
-                onClick={() => setAiModel('3.0')}
-                className={cn("flex-1 py-1.5 text-sm rounded-md transition-all font-medium", aiModel === '3.0' ? "bg-slate-700 text-white" : "text-slate-400 hover:text-slate-200")}
-              >
-                Gemini 3.0
-              </button>
-            </div>
             <div className="flex bg-black/20 rounded-lg p-1 mt-2">
               <button 
                 onClick={() => setMode('payload')}
@@ -1063,14 +1048,21 @@ function SettingsTab() {
   const [hfToken, setHfToken] = useState('');
   const [ghToken, setGhToken] = useState('');
   const [geminiToken, setGeminiToken] = useState('');
+  const [geminiModel, setGeminiModel] = useState<'2.5' | '3.0'>('2.5');
   const [status, setStatus] = useState('');
 
+  useEffect(() => {
+    fetch('/api/tokens/status').then(res => res.json()).then(data => {
+      if (data.gemini_model) setGeminiModel(data.gemini_model);
+    }).catch(()=>{});
+  }, []);
+
   const handleSave = async () => {
     try {
       const res = await fetch('/api/tokens', {
         method: 'POST',
         headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({ hf_token: hfToken, github_token: ghToken, gemini_token: geminiToken })
+        body: JSON.stringify({ hf_token: hfToken, github_token: ghToken, gemini_token: geminiToken, gemini_model: geminiModel })
       });
       const data = await res.json();
       setStatus(data.message || 'Saved securely to in-memory state.');
@@ -1092,6 +1084,23 @@ function SettingsTab() {
 
       <Card title="API Tokens" icon={<ShieldCheck size={18} />}>
         <div className="flex flex-col gap-4">
+          <div>
+            <label className="block text-sm font-medium text-slate-400 mb-1.5 px-1">Gemini AI Model Generation Version</label>
+            <div className="flex bg-black/20 border border-white/10 rounded-xl p-1 w-full gap-1">
+              <button 
+                onClick={() => setGeminiModel('2.5')}
+                className={cn("flex-1 py-2 text-sm rounded-lg transition-all font-medium", geminiModel === '2.5' ? "bg-indigo-500 text-white shadow-md shadow-indigo-500/20" : "text-slate-400 hover:text-slate-200")}
+              >
+                Gemini 2.5 Pro
+              </button>
+              <button 
+                onClick={() => setGeminiModel('3.0')}
+                className={cn("flex-1 py-2 text-sm rounded-lg transition-all font-medium", geminiModel === '3.0' ? "bg-indigo-500 text-white shadow-md shadow-indigo-500/20" : "text-slate-400 hover:text-slate-200")}
+              >
+                Gemini 3.0 Pro
+              </button>
+            </div>
+          </div>
           <div>
             <label className="block text-sm font-medium text-slate-400 mb-1.5 px-1">Hugging Face Token</label>
             <input 
@@ -1187,8 +1196,8 @@ function DeployTab() {
               onClick={() => setProvider('github')}
               className={cn("p-4 rounded-2xl border transition-all text-left", provider === 'github' ? "glass border-indigo-500/50" : "bg-black/20 border-white/5 hover:border-white/10")}
             >
-              <h3 className="font-semibold text-white">GitHub Pages</h3>
-              <p className="text-xs text-slate-400 mt-1">Deploy JS background clients to GH Repos.</p>
+              <h3 className="font-semibold text-white">GitHub Actions</h3>
+              <p className="text-xs text-slate-400 mt-1">Deploy Python background clients to GH Repos.</p>
             </button>
           </div>