feat: token in env

app.py

@@ -107,9 +107,20 @@ def compute_coverage_score(eval_data):
 def get_llm_feedback(yaml_content, api_token=None):
     """
     Get feedback on the eval card from Groq's LLM
+    Uses GROQ_API_KEY from environment variables if no token is provided
     """
+    import os
+    import requests
+    from dotenv import load_dotenv
+    
+    # Load environment variables from .env file if it exists
+    load_dotenv()
+    
+    # Use provided token or get from environment
+    api_token = api_token or os.environ.get("GROQ_API_KEY")
+    
     if not api_token:
-        return "API token is required for LLM feedback."
+        return "API token is required for LLM feedback. Please set the GROQ_API_KEY environment variable or provide a token."
 
     try:
         headers = {
@@ -277,13 +288,16 @@ def upload_file(file):
     except Exception as e:
         return f"Error processing file: {str(e)}", None
 
-def get_feedback(yaml_content, api_token):
+def get_feedback(yaml_content):
     """Get LLM feedback on the eval card"""
     if not yaml_content:
         return "Please upload or paste a YAML file first."
     
+    # Use provided token or get from environment
+    api_token = os.environ.get("GROQ_API_KEY")
+    
     if not api_token:
-        return "Please provide an API token for the LLM service."
+        return "Please provide an API token or set the GROQ_API_KEY environment variable."
     
     feedback = get_llm_feedback(yaml_content, api_token)
     return feedback
@@ -360,7 +374,7 @@ with gr.Blocks(title="Evaluation Card Repository") as app:
                     
                     load_template_btn = gr.Button("Load Template")
                     
-                    api_token = gr.Textbox(label="API Token (for LLM feedback)", type="password")
+                    # api_token = gr.Textbox(label="API Token (for LLM feedback)", type="password")
                     
                     with gr.Row():
                         get_feedback_btn = gr.Button("Get LLM Feedback")
@@ -407,7 +421,7 @@ with gr.Blocks(title="Evaluation Card Repository") as app:
     
     get_feedback_btn.click(
         fn=get_feedback,
-        inputs=[yaml_display, api_token],
+        inputs=[yaml_display],
         outputs=[feedback_display]
     )
     

requirement.txt

@@ -2,3 +2,4 @@ gradio>=4.0.0
 pyyaml>=6.0
 pandas>=2.0.0
 requests>=2.31.0
+python-dotenv>=1.0.0