Update app.py

app.py

@@ -1,4 +1,5 @@
-import google.generativeai as genai
+# import google.generativeai as genai
+from groq import Groq
 import gradio as gr
 import os
 import json
@@ -10,15 +11,23 @@ import re
 
 # Set your API key
 # Or use `os.getenv('GOOGLE_API_KEY')` to fetch an environment variable.
-GOOGLE_API_KEY=os.getenv('GEMINI_KEY')
+# GOOGLE_API_KEY=os.getenv('GEMINI_KEY')
 
-genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-pro')
+# genai.configure(api_key=GOOGLE_API_KEY)
+# model = genai.GenerativeModel('gemini-pro')
 
 # Select the PaLM 2 model
 # model = 'models/text-bison-001'
 
 
+api1 = os.getenv("Groq_key")
+
+apis = [
+    api1,
+    # api1,
+]
+
+
 # contx = """Act as a friendly personal assistant. 
 # You help users manage their daily tasks by providing them with a variety of features.
 # Below are the list of features you have: \n
@@ -65,8 +74,31 @@ def responsenew(data):
     #     print(query)
     # except json.JSONDecodeError as e:
     #     print("Invalid JSON:", e)
-    response = model.generate_content(query)
-    print(f"\n{response.text}")
+    while True:
+        for api in apis:
+            client = Groq(
+                    api_key=api,
+                )  # Configure the model with the API key
+            # query = st.text_input("Enter your query")
+            # prmptquery= f"Answer this query with wisdom, love and compassion, in context to bhagwat geeta, provide references of shloks from chapters of bhagwat geeta which is relevant to the query. Query= {query}"
+            try:
+                response = client.chat.completions.create(
+                messages=[
+                    {
+                        "role": "user",
+                        "content": query,
+                    }
+                ],
+                model="mixtral-8x7b-32768",
+                )
+                answer = response.choices[0].message.content
+            except Exception as e:
+                print(f"API call failed for: {e}")
+            if answer:
+                break
+        if answer:
+                break
+    print(f"\n{answer}")
     if query is not None:
         if "remind me" in query:
             values = getValues(query)
@@ -239,7 +271,7 @@ def responsenew(data):
             }
         else:
             respo = {
-                "message": response.text,
+                "message": answer,
                 "action": "nothing",
                 "function": "nothing",
             }