Update app.py

app.py

@@ -3,11 +3,27 @@ st.set_page_config(page_title="B2B Problem Solver", layout="wide")
 
 import requests
 import os
+import json
 from bs4 import BeautifulSoup
 
-# Configure OpenAI API (using OPENAI_API_KEY)
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-OPENAI_ENDPOINT = "https://api.openai.com/v1/chat/completions"
+# API Configuration options
+API_OPTIONS = {
+    "OpenAI": {
+        "endpoint": "https://api.openai.com/v1/chat/completions",
+        "env_key": "OPENAI_API_KEY",
+        "models": ["gpt-4", "gpt-3.5-turbo"]
+    },
+    "Anthropic": {
+        "endpoint": "https://api.anthropic.com/v1/messages",
+        "env_key": "ANTHROPIC_API_KEY",
+        "models": ["claude-3-opus-20240229", "claude-3-sonnet-20240229", "claude-3-haiku-20240307"]
+    },
+    "Azure OpenAI": {
+        "endpoint": os.getenv("AZURE_OPENAI_ENDPOINT", "https://your-resource-name.openai.azure.com/openai/deployments/your-deployment-name/chat/completions?api-version=2023-05-15"),
+        "env_key": "AZURE_OPENAI_API_KEY",
+        "models": ["your-deployment-name"]
+    }
+}
 
 # Define common business problem categories
 PROBLEM_CATEGORIES = [
@@ -25,6 +41,14 @@ PROBLEM_CATEGORIES = [
     "Other (Please Specify)"
 ]
 
+# Get API settings from session state or initialize them
+if 'api_provider' not in st.session_state:
+    st.session_state.api_provider = "OpenAI"
+if 'api_model' not in st.session_state:
+    st.session_state.api_model = API_OPTIONS["OpenAI"]["models"][0]
+if 'api_key' not in st.session_state:
+    st.session_state.api_key = os.getenv(API_OPTIONS["OpenAI"]["env_key"], "")
+
 def scrape_company_info(url):
     """Scrapes business information from the given company website URL."""
     if not url.startswith("http"):
@@ -52,7 +76,7 @@ def scrape_company_info(url):
         return None
 
 def generate_problem_solution(business_info, problem_category, problem_description):
-    """Generates a practical solution for the specified business problem using OpenAI GPT-4."""
+    """Generates a practical solution for the specified business problem using the configured AI provider."""
     prompt = f"""As a senior B2B growth strategist with extensive experience helping sales and marketing executives overcome business challenges, provide a practical solution for the following scenario:
 
 COMPANY INFORMATION:
@@ -79,13 +103,35 @@ Please provide:
 
 Format your response in clear markdown with headers and bullet points, keeping it concise, practical, and immediately actionable for busy executives."""
     
+    provider = st.session_state.api_provider
+    model = st.session_state.api_model
+    api_key = st.session_state.api_key
+    
+    if not api_key:
+        return f"Error: Missing API key for {provider}. Please configure your API key in the settings."
+    
+    try:
+        if provider == "OpenAI":
+            return call_openai_api(prompt, model, api_key)
+        elif provider == "Anthropic":
+            return call_anthropic_api(prompt, model, api_key)
+        elif provider == "Azure OpenAI":
+            return call_azure_openai_api(prompt, model, api_key)
+        else:
+            return "Error: Invalid API provider selected"
+    except Exception as e:
+        error_message = str(e)
+        return f"API Error: {error_message}\n\nPlease check your API configuration in the settings panel."
+
+def call_openai_api(prompt, model, api_key):
+    """Call the OpenAI API to generate a response."""
     headers = {
-        "Authorization": f"Bearer {OPENAI_API_KEY}",
+        "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
     
     data = {
-        "model": "gpt-4",  # Using OpenAI GPT-4 model
+        "model": model,
         "messages": [
             {"role": "system", "content": "You are a senior B2B growth strategist who specializes in solving critical business challenges for sales and marketing executives."},
             {"role": "user", "content": prompt}
@@ -94,18 +140,82 @@ Format your response in clear markdown with headers and bullet points, keeping i
         "max_tokens": 3500
     }
     
-    try:
-        response = requests.post(OPENAI_ENDPOINT, json=data, headers=headers)
-        response.raise_for_status()
-        return response.json()["choices"][0]["message"]["content"]
-    except Exception as e:
-        st.error(f"API Error: {str(e)}")
-        return None
+    response = requests.post(API_OPTIONS["OpenAI"]["endpoint"], json=data, headers=headers)
+    response.raise_for_status()
+    return response.json()["choices"][0]["message"]["content"]
+
+def call_anthropic_api(prompt, model, api_key):
+    """Call the Anthropic API to generate a response."""
+    headers = {
+        "x-api-key": api_key,
+        "anthropic-version": "2023-06-01",
+        "content-type": "application/json"
+    }
+    
+    data = {
+        "model": model,
+        "messages": [
+            {"role": "user", "content": prompt}
+        ],
+        "max_tokens": 3500
+    }
+    
+    response = requests.post(API_OPTIONS["Anthropic"]["endpoint"], json=data, headers=headers)
+    response.raise_for_status()
+    return response.json()["content"][0]["text"]
+
+def call_azure_openai_api(prompt, model, api_key):
+    """Call the Azure OpenAI API to generate a response."""
+    headers = {
+        "api-key": api_key,
+        "Content-Type": "application/json"
+    }
+    
+    data = {
+        "messages": [
+            {"role": "system", "content": "You are a senior B2B growth strategist who specializes in solving critical business challenges for sales and marketing executives."},
+            {"role": "user", "content": prompt}
+        ],
+        "temperature": 0.7,
+        "max_tokens": 3500
+    }
+    
+    response = requests.post(API_OPTIONS["Azure OpenAI"]["endpoint"], json=data, headers=headers)
+    response.raise_for_status()
+    return response.json()["choices"][0]["message"]["content"]
 
 # Streamlit UI
 st.title("🚀 B2B Problem Solver")
 st.markdown("### Get practical solutions to common business problems hindering your growth")
 
+# API Configuration in sidebar
+with st.sidebar:
+    st.header("API Settings")
+    st.session_state.api_provider = st.selectbox(
+        "Select AI Provider", 
+        options=list(API_OPTIONS.keys()),
+        index=list(API_OPTIONS.keys()).index(st.session_state.api_provider)
+    )
+    
+    st.session_state.api_model = st.selectbox(
+        "Select Model",
+        options=API_OPTIONS[st.session_state.api_provider]["models"],
+        index=min(API_OPTIONS[st.session_state.api_provider]["models"].index(st.session_state.api_model) 
+                  if st.session_state.api_model in API_OPTIONS[st.session_state.api_provider]["models"] else 0,
+                 len(API_OPTIONS[st.session_state.api_provider]["models"]) - 1)
+    )
+    
+    api_key_env = API_OPTIONS[st.session_state.api_provider]["env_key"]
+    api_key_input = st.text_input(
+        f"API Key (or set {api_key_env} environment variable)",
+        type="password",
+        value=st.session_state.api_key
+    )
+    if api_key_input:
+        st.session_state.api_key = api_key_input
+    
+    st.info(f"Current status: {'✅ API key configured' if st.session_state.api_key else '❌ API key missing'}")
+
 with st.expander("💡 How it works", expanded=True):
     st.markdown("""
     1. Enter your company details or website URL
@@ -133,6 +243,8 @@ if input_method == "Enter company details manually":
     if st.button("Generate Solution"):
         if not all([company_name, industry, company_description, problem_description]):
             st.error("Please fill in all required fields marked with *")
+        elif not st.session_state.api_key:
+            st.error(f"Please configure your {st.session_state.api_provider} API key in the sidebar")
         else:
             with st.spinner("Analyzing your business challenge and crafting a solution..."):
                 business_info = f"""
@@ -143,17 +255,24 @@ Company Size: {team_size}
 """
                 solution = generate_problem_solution(business_info, problem_category, problem_description)
                 if solution:
-                    st.markdown("## Your Customized Solution Plan")
-                    st.markdown(solution)
-                    
-                    with st.expander("Want expert help implementing this solution?"):
-                        st.markdown("""
-                        Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
+                    if not solution.startswith("Error") and not solution.startswith("API Error"):
+                        st.markdown("## Your Customized Solution Plan")
+                        st.markdown(solution)
                         
-                        **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
-                        """)
-                        st.text_input("Your Email")
-                        st.button("Request Consultation")
+                        with st.expander("Want expert help implementing this solution?"):
+                            st.markdown("""
+                            Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
+                            
+                            **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
+                            """)
+                            contact_email = st.text_input("Your Email")
+                            if st.button("Request Consultation"):
+                                if contact_email:
+                                    st.success("Thank you! We'll be in touch shortly to schedule your consultation.")
+                                else:
+                                    st.warning("Please enter your email address.")
+                    else:
+                        st.error(solution)
 
 else:
     website_url = st.text_input("Enter your company website URL*", placeholder="e.g., www.example.com")
@@ -169,23 +288,32 @@ else:
     if st.button("Generate Solution"):
         if not all([website_url, problem_description]):
             st.error("Please fill in all required fields marked with *")
+        elif not st.session_state.api_key:
+            st.error(f"Please configure your {st.session_state.api_provider} API key in the sidebar")
         else:
             with st.spinner("Extracting company details and generating your solution..."):
                 company_info = scrape_company_info(website_url)
                 if company_info:
                     solution = generate_problem_solution(company_info, problem_category, problem_description)
                     if solution:
-                        st.markdown("## Your Customized Solution Plan")
-                        st.markdown(solution)
-                        
-                        with st.expander("Want expert help implementing this solution?"):
-                            st.markdown("""
-                            Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
+                        if not solution.startswith("Error") and not solution.startswith("API Error"):
+                            st.markdown("## Your Customized Solution Plan")
+                            st.markdown(solution)
                             
-                            **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
-                            """)
-                            st.text_input("Your Email")
-                            st.button("Request Consultation")
+                            with st.expander("Want expert help implementing this solution?"):
+                                st.markdown("""
+                                Our growth agency specializes in helping B2B companies implement these solutions and achieve measurable results.
+                                
+                                **Schedule a free 30-minute consultation** to discuss how we can help you overcome this challenge and accelerate your growth.
+                                """)
+                                contact_email = st.text_input("Your Email")
+                                if st.button("Request Consultation"):
+                                    if contact_email:
+                                        st.success("Thank you! We'll be in touch shortly to schedule your consultation.")
+                                    else:
+                                        st.warning("Please enter your email address.")
+                        else:
+                            st.error(solution)
 
 # Add footer with branding
 st.markdown("---")