Update app.py

app.py

@@ -29,7 +29,7 @@ def generate_preliminary_plan(cluster_data):
         response = requests.post(
             'https://openrouter.ai/api/v1/chat/completions',
             headers={
-                'Authorization': f'Bearer {OPENROUTER_API_KEY}',
+                'Authorization': f'Bearer {cluster_data["openrouter_key"]}',
                 'HTTP-Referer': 'http://127.0.0.1:5001',
                 'X-Title': 'Blog Generator'
             },
@@ -90,7 +90,7 @@ Create the preliminary plan."""
         logger.error(f"Error in generate_preliminary_plan: {e}")
         raise
 
-def do_research(plan):
+def do_research(plan, openrouter_key):
     logger.info("Doing research...")
     try:
         # Extract key points from plan to create search queries
@@ -109,7 +109,7 @@ def do_research(plan):
                 
             # Get research results including both web content and AI analysis
             # Using 5 sites per point for more comprehensive research
-            results = research_topic(point, num_sites=5)
+            results = research_topic(point, num_sites=5, openrouter_key=openrouter_key)
             
             if results['success']:
                 all_research.append({
@@ -163,7 +163,7 @@ def generate_blog():
 
                 # 3. Do research
                 logger.info("Doing research...")
-                research = do_research(plan)
+                research = do_research(plan, OPENROUTER_API_KEY)
 
                 # 4. Create detailed plan
                 logger.info("Creating detailed plan...")
@@ -295,7 +295,7 @@ def generate_from_csv():
                 plan = generate_preliminary_plan(cluster_data)
                 
                 # Do research
-                research = do_research(plan)
+                research = do_research(plan, OPENROUTER_API_KEY)
                 
                 # Create detailed plan
                 detailed_plan = blog_gen.create_detailed_plan(cluster_data, plan, research)
@@ -360,15 +360,18 @@ def generate_from_csv_text():
     try:
         logger.info("Starting blog generation process for multiple clusters...")
         
-        # Get CSV content from request JSON
+        # Get CSV content and OpenRouter API key from request JSON
         data = request.get_json()
         if not data or 'csv_content' not in data:
             return jsonify({'error': 'No CSV content provided'}), 400
+        if 'openrouter_key' not in data:
+            return jsonify({'error': 'OpenRouter API key is required'}), 400
             
         csv_content = data['csv_content']
+        openrouter_key = data['openrouter_key']
         
-        # Initialize handlers
-        blog_gen = BlogGenerator(OPENAI_API_KEY, OPENROUTER_API_KEY)
+        # Initialize handlers with the provided OpenRouter key
+        blog_gen = BlogGenerator(OPENAI_API_KEY, openrouter_key)
         csv_handler = CSVHandler()
         
         # Process the CSV text
@@ -384,13 +387,16 @@ def generate_from_csv_text():
             try:
                 logger.info(f"Processing cluster with primary keyword: {cluster_data['Primary Keyword']}")
                 
+                # Add OpenRouter key to cluster_data for use in functions
+                cluster_data['openrouter_key'] = openrouter_key
+                
                 # Generate preliminary plan
                 logger.info("Generating preliminary plan...")
                 plan = generate_preliminary_plan(cluster_data)
 
                 # Do research
                 logger.info("Doing research...")
-                research = do_research(plan)
+                research = do_research(plan, openrouter_key)
 
                 # Create detailed plan
                 logger.info("Creating detailed plan...")
@@ -459,4 +465,4 @@ def generate_from_csv_text():
         return jsonify({'error': str(e)}), 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+    app.run(host='127.0.0.1', port=5001, debug=True)