Add database update functionality with admin password protection and logging

app.py

@@ -20,8 +20,8 @@ from svglib.svglib import svg2rlg
 dataset = load_dataset("smitathkr1/organic_reactions_corrected")
 df = dataset['train'].to_pandas()
 
-# Precompute unique values for autocomplete
-reaction_names = df['corrected_name'].unique().tolist()
+# Global variable to store the last AI fix for potential database update
+last_ai_fix = None
 all_reactants = []
 all_products = []
 
@@ -269,23 +269,53 @@ def fix_reaction_with_gemini(reaction_name, api_key):
         import json
         gemini_data = json.loads(response_text)
         
-        # Update the global df (this won't persist, but shows the updated data)
+        # Store the updated data globally for potential database update
+        global last_ai_fix
+        last_ai_fix = {
+            'reaction_name': reaction_name,
+            'updated_data': gemini_data,
+            'timestamp': str(pd.Timestamp.now())
+        }
+        
+        return f"✅ **AI Fix Completed for '{reaction_name}'**\n\n**Updated Data:**\n- **Name:** {gemini_data.get('reaction name', 'N/A')}\n- **Reactants:** {', '.join(gemini_data.get('reactants', []))}\n- **Reagents:** {', '.join(gemini_data.get('reagents', []))}\n- **Products:** {', '.join(gemini_data.get('products', []))}\n- **Description:** {gemini_data.get('description', '')[:200]}...\n\n💡 **To save this fix to the database, enter the admin password below and click 'Update Database'.**"
+            
+    except Exception as e:
+        return f"❌ Error calling Gemini API: {str(e)}"
+
+def update_database_with_ai_fix(password):
+    if password != "Vvn@#411037":
+        return "❌ Incorrect password. Database update denied."
+    
+    global last_ai_fix
+    if not last_ai_fix:
+        return "❌ No recent AI fix to save. Please fix a reaction first."
+    
+    try:
+        # Update the global df
         global df
-        mask = df['corrected_name'].str.lower() == reaction_name.lower()
+        mask = df['corrected_name'].str.lower() == last_ai_fix['reaction_name'].lower()
         if mask.any():
             idx = df[mask].index[0]
-            df.at[idx, 'corrected_name'] = gemini_data.get('reaction name', reaction_name)
-            df.at[idx, 'general_reactants'] = ', '.join(gemini_data.get('reactants', []))
-            df.at[idx, 'general_reagents'] = ', '.join(gemini_data.get('reagents', []))
-            df.at[idx, 'general_products'] = ', '.join(gemini_data.get('products', []))
-            df.at[idx, 'description'] = gemini_data.get('description', df.at[idx, 'description'])
+            df.at[idx, 'corrected_name'] = last_ai_fix['updated_data'].get('reaction name', last_ai_fix['reaction_name'])
+            df.at[idx, 'general_reactants'] = ', '.join(last_ai_fix['updated_data'].get('reactants', []))
+            df.at[idx, 'general_reagents'] = ', '.join(last_ai_fix['updated_data'].get('reagents', []))
+            df.at[idx, 'general_products'] = ', '.join(last_ai_fix['updated_data'].get('products', []))
+            df.at[idx, 'description'] = last_ai_fix['updated_data'].get('description', df.at[idx, 'description'])
             
-            return f"Successfully updated reaction '{reaction_name}' with AI-generated data:\n\n**New Name:** {gemini_data.get('reaction name')}\n**Reactants:** {', '.join(gemini_data.get('reactants', []))}\n**Reagents:** {', '.join(gemini_data.get('reagents', []))}\n**Products:** {', '.join(gemini_data.get('products', []))}\n**Description:** {gemini_data.get('description', '')[:200]}..."
+            # Log the update
+            log_entry = f"[{last_ai_fix['timestamp']}] Database updated: '{last_ai_fix['reaction_name']}' -> '{last_ai_fix['updated_data'].get('reaction name', last_ai_fix['reaction_name'])}'\n"
+            with open('database_updates.log', 'a') as f:
+                f.write(log_entry)
+            
+            # Clear the last fix
+            last_ai_fix = None
+            
+            return f"✅ **Database Updated Successfully!**\n\nThe reaction has been permanently updated in the current session. Changes logged to 'database_updates.log'.\n\n**Note:** Since this is a Hugging Face Space, the changes are session-only. To make permanent changes, please update your local dataset and re-upload."
         else:
-            return f"Reaction '{reaction_name}' not found in database."
+            return f"❌ Reaction '{last_ai_fix['reaction_name']}' not found in database."
             
     except Exception as e:
-        return f"Error calling Gemini API: {str(e)}"
+        return f"❌ Error updating database: {str(e)}"
 
 def get_autocomplete_reactants(query):
     if not query:
@@ -346,6 +376,16 @@ with gr.Blocks(title="Organic Reactions Search") as demo:
         
         fix_button.click(fix_reaction_with_gemini, inputs=[reaction_to_fix, api_key_input], outputs=ai_status)
         
+        # Database update section
+        gr.Markdown("---")
+        gr.Markdown("**Database Update (Admin Only):**")
+        with gr.Row():
+            admin_password = gr.Textbox(label="Admin Password", type="password", placeholder="Enter admin password to update database")
+            update_db_button = gr.Button("Update Database", variant="secondary")
+        
+        update_status = gr.Markdown(label="Update Status")
+        update_db_button.click(update_database_with_ai_fix, inputs=[admin_password], outputs=update_status)
+        
         gr.Markdown("---")
         gr.Markdown("**Database Table:**")