Update app.py

app.py

@@ -7,12 +7,14 @@ from langchain_groq import ChatGroq
 from langsmith import traceable
 from typing import TypedDict
 
-# Load API keys from environment variables
+# Load API Keys (Set in Hugging Face Spaces)
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 LANGSMITH_API_KEY = os.getenv("LANGSMITH_API_KEY")
 
-os.environ["LANGCHAIN_TRACING_V2"] = "true"
-os.environ["LANGCHAIN_API_KEY"] = LANGSMITH_API_KEY
+# Ensure API Keys are set
+if not GROQ_API_KEY or not LANGSMITH_API_KEY:
+    st.error("⚠️ Please set GROQ_API_KEY and LANGSMITH_API_KEY in your environment variables.")
+    st.stop()
 
 # Initialize Groq LLM
 llm = ChatGroq(groq_api_key=GROQ_API_KEY, model_name="mixtral-8x7b-32768")
@@ -61,44 +63,48 @@ def generate_test_cases(data):
 def make_code_review_graph():
     """Create a LangGraph workflow for automated code reviews"""
     graph_workflow = StateGraph(State)
-    
+
     graph_workflow.add_node("code_review", code_review)
     graph_workflow.add_node("improvement_suggestions", improvement_suggestions)
     graph_workflow.add_node("generate_documentation", generate_documentation)
     graph_workflow.add_node("generate_test_cases", generate_test_cases)
-    
+
     graph_workflow.add_edge(START, "code_review")
     graph_workflow.add_edge("code_review", "improvement_suggestions")
     graph_workflow.add_edge("improvement_suggestions", "generate_documentation")
     graph_workflow.add_edge("generate_documentation", "generate_test_cases")
     graph_workflow.add_edge("generate_test_cases", END)
-    
+
     return graph_workflow.compile()
 
 # Streamlit UI
-st.title("🛠 AI-Powered Code Review with LangSmith Debugging")
-st.write("Automate your code reviews using Groq AI, LangGraph, and LangSmith.")
+st.title("🛠 AI-Powered Code Review with LangGraph & LangSmith")
+st.write("Analyze and improve your code using AI-based feedback, suggestions, documentation, and test cases.")
 
-code_snippet = st.text_area("Paste your code snippet for review:", height=200)
+# Input Field
+code_snippet = st.text_area("📌 Paste your code snippet below:", height=200)
 
-if st.button("Review Code"):
+if st.button("🔍 Review Code"):
     if not code_snippet.strip():
-        st.warning("⚠️ Please enter a code snippet.")
+        st.warning("⚠️ Please enter a valid code snippet.")
     else:
         try:
             review_agent = make_code_review_graph()
             result = review_agent.invoke({"code_snippet": code_snippet})
-            
+
+            # Display Results with Clear Formatting
             st.subheader("💡 Review Comments")
-            st.text_area("", result.get("review_comments", ""), height=150)
-            
+            st.write(result["review_comments"])
+
             st.subheader("🔧 Suggested Improvements")
-            st.text_area("", result.get("suggestions", ""), height=150)
-            
+            st.write(result["suggestions"])
+
             st.subheader("📖 Generated Documentation")
-            st.text_area("", result.get("documentation", ""), height=150)
-            
+            st.code(result["documentation"], language="python")
+
             st.subheader("🧪 Suggested Test Cases")
-            st.text_area("", result.get("test_cases", ""), height=150)
+            st.code(result["test_cases"], language="python")
+
         except Exception as e:
-            st.error(f"⚠️ Error: {str(e)}\n{traceback.format_exc()}")
+            st.error(f"⚠️ Error: {str(e)}")
+            st.text(traceback.format_exc())