Update app.py

app.py

@@ -78,6 +78,16 @@ if not GROQ_API_KEY:
     st.stop()
 client = Groq(api_key=GROQ_API_KEY)
 
+# --- Blackbox AI Agent Setup ---
+BLACKBOX_API_KEY = os.environ.get("BLACKBOX_API_KEY")
+if not BLACKBOX_API_KEY:
+    st.error("BLACKBOX_API_KEY environment variable not set. Please set it in your Hugging Face Space secrets.")
+    st.stop()
+
+# Chat history management
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+
 def groq_api_call(prompt):
     chat_completion = client.chat.completions.create(
         messages=[{"role": "user", "content": prompt}],
@@ -85,6 +95,13 @@ def groq_api_call(prompt):
     )
     return chat_completion.choices[0].message.content
 
+def blackbox_ai_call(messages):
+    # This is a placeholder for actual Blackbox AI API call using BLACKBOX_API_KEY
+    # For demonstration, we simulate a response by echoing last user message
+    last_user_message = messages[-1]["content"] if messages else ""
+    response = f"Blackbox AI response to: {last_user_message}"
+    return response
+
 def get_diff_html(original, modified):
     original_lines = original.splitlines()
     modified_lines = modified.splitlines()
@@ -246,6 +263,8 @@ if "timeline" not in st.session_state:
     st.session_state.timeline = []
 if "suggestions" not in st.session_state:
     st.session_state.suggestions = []
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
 
 col1, col2 = st.columns([2, 3], gap="large")
 
@@ -313,24 +332,19 @@ with col2:
                         st.session_state.suggestions = suggestions
                     st.success("Agentic workflow complete. See timeline below.")
 
-    if st.session_state.timeline:
-        for i, step in enumerate(st.session_state.timeline):
-            with st.expander(f"Step {i+1}: {step['step']}"):
-                st.markdown(f"<b>{step['description']}</b>", unsafe_allow_html=True)
-                if step['step'] in ["Refactor", "Test Generation"]:
-                    if i > 0:
-                        prev_code = st.session_state.timeline[i-1]['code']
-                        diff_html = get_diff_html(prev_code, step['code'])
-                        st.markdown("**Side-by-Side Diff:**")
-                        st.components.v1.html(diff_html, height=400, scrolling=True)
-                if step['step'] in ["Refactor", "Test Generation"]:
-                    st.markdown("**Code Output:**")
-                    st.code(step['output'], language=programming_language.lower())
-                else:
-                    st.markdown("**Output:**")
-                    st.write(step['output'])
-    else:
-        st.info("Run the agentic workflow to see step-by-step results, explanations, and code evolution.")
+    # Chatbox with history using Blackbox AI agent
+    st.subheader("Chat with Blackbox AI Agent")
+    user_input = st.text_input("Enter your message:", key="chat_input")
+    if user_input:
+        st.session_state.chat_history.append({"role": "user", "content": user_input})
+        response = blackbox_ai_call(st.session_state.chat_history)
+        st.session_state.chat_history.append({"role": "assistant", "content": response})
+
+    for chat in st.session_state.chat_history:
+        if chat["role"] == "user":
+            st.markdown(f"**You:** {chat['content']}")
+        else:
+            st.markdown(f"**Blackbox AI:** {chat['content']}")
 
 st.markdown("---")
-st.markdown('<div style="text-align: center; color: #22304a; font-size: 1rem; margin-top: 2em;">Powered by <b>BLACKBOX.AI</b></div>', unsafe_allow_html=True)
+st.markdown('<div style="text-align: center; color: #22304a; font-size: 1rem; margin-top: 2em;">Powered by <b>BLACKBOX.AI</b></div>', unsafe_allow_html=True)