Update app.py

app.py

@@ -13,6 +13,9 @@ from concurrent.futures import ThreadPoolExecutor, as_completed
 from tenacity import retry, wait_exponential, stop_after_attempt, retry_if_exception_type
 import threading
 from anthropic import Anthropic
+import google.generativeai as genai
+
+
 
 load_dotenv()
 
@@ -29,6 +32,7 @@ anthropic_client = Anthropic(
     api_key=os.getenv('ANTHROPIC_API_KEY')
 )
 
+genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 MAX_CONCURRENT_CALLS = 5
 semaphore = threading.Semaphore(MAX_CONCURRENT_CALLS)
@@ -97,6 +101,16 @@ def get_model_response(question, options, prompt_template, model_name):
                 )
                 response_text =  response.content[0].text
 
+            elif provider == "google":
+                model = genai.GenerativeModel(
+                model_name=model_config["model_id"]
+                )
+
+                chat_session = model.start_chat(
+                history=[]
+                )
+                response_text = chat_session.send_message(prompt).text
+
             json_match = re.search(r'\{.*\}', response_text, re.DOTALL)
             if not json_match:
                 return f"Error: Invalid response format", response_text
@@ -296,7 +310,7 @@ Important:
         ).mark_bar().encode(
             x=alt.X('index:N', title=None, axis=None), 
             y=alt.Y('value:Q', title='Accuracy', scale=alt.Scale(domain=[0, 1])),
-            color='index:N',
+            color=alt.Color('index:N', scale=alt.Scale(scheme='blues')), 
             tooltip=['index:N', 'value:Q']
         ).properties(
             height=300,
@@ -307,6 +321,7 @@ Important:
                 "dy": 20              
             }
         )
+
         st.altair_chart(accuracy_chart, use_container_width=True)
 
     if st.session_state.all_results: