Update app.py

app.py

@@ -4,7 +4,13 @@ import os
 import json
 import time
 
-client = groq.Groq()
+# Initialize the Groq client with the API key from Hugging Face secrets
+api_key = os.environ.get("GROQ_API_KEY")
+if not api_key:
+    st.error("No Groq API key found. Please add your GROQ_API_KEY to the Hugging Face Space secrets.")
+    st.stop()
+
+client = groq.Groq(api_key=api_key)
 
 def make_api_call(messages, max_tokens, is_final_answer=False):
     for attempt in range(3):
@@ -25,57 +31,7 @@ def make_api_call(messages, max_tokens, is_final_answer=False):
                     return {"title": "Error", "content": f"Failed to generate step after 3 attempts. Error: {str(e)}", "next_action": "final_answer"}
             time.sleep(1)  # Wait for 1 second before retrying
 
-def generate_response(prompt):
-    messages = [
-        {"role": "system", "content": """You are an expert AI assistant that explains your reasoning step by step. For each step, provide a title that describes what you're doing in that step, along with the content. Decide if you need another step or if you're ready to give the final answer. Respond in JSON format with 'title', 'content', and 'next_action' (either 'continue' or 'final_answer') keys. USE AS MANY REASONING STEPS AS POSSIBLE. AT LEAST 3. BE AWARE OF YOUR LIMITATIONS AS AN LLM AND WHAT YOU CAN AND CANNOT DO. IN YOUR REASONING, INCLUDE EXPLORATION OF ALTERNATIVE ANSWERS. CONSIDER YOU MAY BE WRONG, AND IF YOU ARE WRONG IN YOUR REASONING, WHERE IT WOULD BE. FULLY TEST ALL OTHER POSSIBILITIES. YOU CAN BE WRONG. WHEN YOU SAY YOU ARE RE-EXAMINING, ACTUALLY RE-EXAMINE, AND USE ANOTHER APPROACH TO DO SO. DO NOT JUST SAY YOU ARE RE-EXAMINING. USE AT LEAST 3 METHODS TO DERIVE THE ANSWER. USE BEST PRACTICES.
-
-Example of a valid JSON response:
-```json
-{
-    "title": "Identifying Key Information",
-    "content": "To begin solving this problem, we need to carefully examine the given information and identify the crucial elements that will guide our solution process. This involves...",
-    "next_action": "continue"
-}```
-"""},
-        {"role": "user", "content": prompt},
-        {"role": "assistant", "content": "Thank you! I will now think step by step following my instructions, starting at the beginning after decomposing the problem."}
-    ]
-    
-    steps = []
-    step_count = 1
-    total_thinking_time = 0
-    
-    while True:
-        start_time = time.time()
-        step_data = make_api_call(messages, 300)
-        end_time = time.time()
-        thinking_time = end_time - start_time
-        total_thinking_time += thinking_time
-        
-        steps.append((f"Step {step_count}: {step_data['title']}", step_data['content'], thinking_time))
-        
-        messages.append({"role": "assistant", "content": json.dumps(step_data)})
-        
-        if step_data['next_action'] == 'final_answer' or step_count > 25: # Maximum of 25 steps to prevent infinite thinking time. Can be adjusted.
-            break
-        
-        step_count += 1
-
-        # Yield after each step for Streamlit to update
-        yield steps, None  # We're not yielding the total time until the end
-
-    # Generate final answer
-    messages.append({"role": "user", "content": "Please provide the final answer based on your reasoning above."})
-    
-    start_time = time.time()
-    final_data = make_api_call(messages, 200, is_final_answer=True)
-    end_time = time.time()
-    thinking_time = end_time - start_time
-    total_thinking_time += thinking_time
-    
-    steps.append(("Final Answer", final_data['content'], thinking_time))
-
-    yield steps, total_thinking_time
+# ... [rest of the code remains the same] ...
 
 def main():
     st.set_page_config(page_title="g1 prototype", page_icon="🧠", layout="wide")
@@ -88,6 +44,20 @@ def main():
     Open source [repository here](https://github.com/bklieger-groq)
     """)
     
+    if not api_key:
+        st.warning("Please add your GROQ_API_KEY to the Hugging Face Space secrets to use this application.")
+        st.markdown("""
+        To add your API key:
+        1. Go to the Settings tab of this Space
+        2. Scroll down to the "Repository secrets" section
+        3. Click on "New secret"
+        4. Set the secret name as `GROQ_API_KEY`
+        5. Paste your Groq API key as the value
+        6. Click "Add secret"
+        7. Rebuild the Space
+        """)
+        st.stop()
+    
     # Text input for user query
     user_query = st.text_input("Enter your query:", placeholder="e.g., How many 'R's are in the word strawberry?")