Update app.py

app.py

@@ -11,8 +11,11 @@ from PIL import Image
 from tools import evaluate_consumption, evaluate_consumption_example
 from retriever import FrugalAI_methods
 
-HF_TOKEN = os.environ.get("HF_TOKEN") 
-API_KEY = os.environ.get("API_KEY") 
+#HF_TOKEN = os.environ.get("HF_TOKEN") 
+try:
+    API_KEY = os.environ.get("API_KEY") 
+except:
+    print("Please provide the Antrhopic API_KEY in the secret variables")
 # Initialize the Hugging Face model
 model = LiteLLMModel(
     model_id="anthropic/claude-3-5-sonnet-latest",
@@ -20,9 +23,9 @@ model = LiteLLMModel(
     api_key=os.environ["API_KEY"]
 )
 #model = InferenceClientModel("deepseek-ai/DeepSeek-R1",max_tokens=500, token=HF_TOKEN)
-headers = {
-    "Authorization": f"Bearer {HF_TOKEN}"
-}
+#headers = {
+ #   "Authorization": f"Bearer {HF_TOKEN}"
+#}
 
 # Initialize the weather tool
 evaluate_consumption = evaluate_consumption()
@@ -58,7 +61,20 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
                 inputs="text",
                 outputs="text",
                 title="Frugalize it!",
-                examples=["What are you capable of? Show me using your custom tool evaluate_consumption_example, and provide examples based on the code in that tool to optimize it for frugality. Use both your tools and web searches if needed., "Here is my code, {code}, please give me frugal alternatives"],
+                examples=[""\nYou are an AI agent that receives Python code from a manager. "
+                "Your task is to test its resource consumption using a custom tool. "
+                "Then, propose frugal alternatives, such as pruning, using your dedicated tool — always use it. "
+                "Additionally, feel free to search the internet for frugal methods and present them to the manager. "
+                "Examples include Knowledge Distillation, Transfer Learning, and others. "
+                "Show me what you can do using your custom tool evaluate_consumption_example, and provide examples based on the code in that tool to optimize it for frugality. "
+                "Use both your tools and web searches if needed.",
+            
+                "\nYou are an AI agent that receives Python code from a manager. "
+                "Your task is to test its resource consumption using a custom tool. "
+                "Then, propose frugal alternatives, such as pruning, using your dedicated tool — always use it. "
+                "Additionally, feel free to search the internet for frugal methods and present them to the manager. "
+                "Examples include Knowledge Distillation, Transfer Learning, and others. "
+                "Here is my code, {code}, please give me frugal alternatives"],
                 description="Share your Python code with this AI agent! It will track its CO2 emissions using CodeCarbon and recommend greener, frugal AI alternatives."
             )