Update app.py

app.py

@@ -2,23 +2,55 @@ import os
 import streamlit as st
 from groq import Groq
 
-client = Groq(
-    api_key=os.environ.get("GROQ_API_KEY"),
-)
 
-query = st.text_input("Enter your query")
-chat_completion = client.chat.completions.create(
+def make_call(api):
+  """Calls the Groq API (assuming API key auth) and handles potential errors."""
+  try:
+    client = Groq(
+            api_key=api,
+        )  # Configure the model with the API key
+    query = st.text_input("Enter your query")
+    prmptquery= f"give the answer of given query in context to bhagwat geeta: {query}"
+    chat_completion = client.chat.completions.create(
     messages=[
         {
             "role": "user",
-            "content": query,
+            "content": prmptquery,
         }
     ],
     model="mixtral-8x7b-32768",
 )
-
-st.write(chat_completion.choices[0].message.content)
-print(chat_completion)
+    # print(response.text)  # Return the response for further processing
+    return chat_completion.choices[0].message.content
+  except Exception as e:
+    print(f"API call failed for key {api_key}: {e}")
+    return None  # Indicate failur
+
+apis = [
+    api1 = os.environ.get("GROQ_API_KEY"),
+    # api1 = os.environ.get("GROQ_API_KEY"),
+    # api1 = os.environ.get("GROQ_API_KEY"),
+    # api1 = os.environ.get("GROQ_API_KEY"),
+    # api1 = os.environ.get("GROQ_API_KEY"),
+    # api1 = os.environ.get("GROQ_API_KEY"),
+]
+
+# Loop indefinitely
+data = None
+while True:  # Loop indefinitely
+    for api in apis:
+        data = make_call(api)
+        if data:  # Check for a successful response
+            st.write(chat_completion.choices[0].message.content)
+            break  # Exit both the for loop and while loop
+        else:
+            st.write(f"Failed to retrieve data from key {api_key}.")
+    if data:  # If a successful response was found, break the outer while loop
+        break
+
+
+
+# print(chat_completion)