Configure AutoStartup.ai to use Gemini API

- Add Gemini 2.0 Flash support to LLMClientManager
- Update default model from o4-mini to gemini-2.0-flash-exp
- Configure Gemini API endpoint and client initialization
- Add environment variable documentation and examples
- Update README with Gemini API setup instructions

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

.env.example

@@ -0,0 +1,8 @@
+# Environment variables for AutoStartup.ai
+OPENAI_MODEL=gemini-2.0-flash-exp
+OPENAI_API_ENDPOINT=https://generativelanguage.googleapis.com/v1beta/openai/
+OPENAI_API_KEY=your_gemini_api_key_here
+
+# Alternative models you can use:
+# OPENAI_MODEL=gpt-4o-mini (requires OpenAI API key)
+# OPENAI_MODEL=gemini-1.5-pro (requires Google AI API key)

README.md

@@ -37,10 +37,28 @@ Transform market problems into viable startup opportunities using AI-driven anal
 ## Installation
 
 ```bash
+# Install dependencies
 pip install -r requirements.txt
+
+# Set up environment variables (choose one):
+# For Gemini API (recommended):
+export OPENAI_MODEL=gemini-2.0-flash-exp
+export OPENAI_API_ENDPOINT=https://generativelanguage.googleapis.com/v1beta/openai/
+export OPENAI_API_KEY=your_gemini_api_key
+
+# For OpenAI API:
+export OPENAI_API_KEY=your_openai_api_key
+
+# Run the application
 python app.py
 ```
 
+## Environment Variables
+
+- `OPENAI_API_KEY`: Your Gemini or OpenAI API key
+- `OPENAI_API_ENDPOINT`: API endpoint (for Gemini: https://generativelanguage.googleapis.com/v1beta/openai/)
+- `OPENAI_MODEL`: Model to use (gemini-2.0-flash-exp, gpt-4o-mini, etc.)
+
 ## Technology Stack
 
 - **Frontend**: Gradio for interactive web interface

app.py

@@ -90,8 +90,12 @@ def generate_and_evaluate(query, ideas_count, random_seed, progress=gr.Progress(
     yield f"Done. Processed {len(ranked_ideas)} ideas in {total_time:.2f}s.", results_df, f"${total_cost:.2f}", ranked_ideas, pd.DataFrame()
     # return
     # --- Compare using generate_and_evaluate2 method ---
-    client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-    ideas = generate_temp_free_idea('ai_scientist/test.json', client, 'gpt-4o-mini', query, ideas_count//4, 4, False)
+    # Create Gemini client for comparison
+    gemini_client = OpenAI(
+        api_key=os.getenv("OPENAI_API_KEY"),
+        base_url=os.getenv("OPENAI_API_ENDPOINT", "https://generativelanguage.googleapis.com/v1beta/openai/")
+    )
+    ideas = generate_temp_free_idea('ai_scientist/test.json', gemini_client, 'gemini-2.0-flash-exp', query, ideas_count//4, 4, False)
     comparison_ranked_ideas, cost = generator.evaluate(ideas, query, progress)
     with open('comparison_ranked_ideas.json', 'w', encoding='utf-8') as f:
         json.dump(comparison_ranked_ideas, f, indent=4, ensure_ascii=False)

generator.py

@@ -81,7 +81,7 @@ class IdeaGenerator():
             json.dump(summary_list, file, indent=4)
         
         self.paper_list = summary_list
-        self.llm.switch_model('o4-mini')
+        self.llm.switch_model('gemini-2.0-flash-exp')
         print("Summary cost: ", self.llm.get_cost())
     
     def get_paper_list(self, num):

utils.py

@@ -57,6 +57,7 @@ class LLMClientManager:
 
     def __init__(self):
         self.openai_client = None
+        self.gemini_client = None
         self.deepseek_client = None
         self.current_client = None
         self.current_model = None
@@ -71,6 +72,16 @@ class LLMClientManager:
                 self.openai_client = OpenAI(api_key=API_KEY)
             self.current_client = self.openai_client
             
+        elif model in ["gemini-2.0-flash-exp", "gemini-1.5-pro", "gemini-1.5-flash"]:
+            if self.gemini_client is None:
+                API_KEY = os.getenv("OPENAI_API_KEY")  # Using same env var for compatibility
+                API_ENDPOINT = os.getenv("OPENAI_API_ENDPOINT", "https://generativelanguage.googleapis.com/v1beta/openai/")
+                self.gemini_client = OpenAI(
+                    api_key=API_KEY,
+                    base_url=API_ENDPOINT
+                )
+            self.current_client = self.gemini_client
+            
         elif model in ["deepseek-chat", "deepseek-reasoner", 'Pro/deepseek-ai/DeepSeek-R1', 'deepseek-r1-250120']:
             self.current_client = self.deepseek_client
         else:
@@ -78,7 +89,7 @@ class LLMClientManager:
 
     def get_response(self, msg, system_message, response_format=None, temperature=1, print_debug=False):
         if self.current_client is None or self.current_model is None:
-            self.switch_model("o4-mini")
+            self.switch_model("gemini-2.0-flash-exp")
         
         msg_history = self.msg_history
         for _ in range(3):