Update app.py

app.py

@@ -9,7 +9,7 @@ from report_generator import (
     generate_diagram_report,
 )
 import cohere
-from cohere.classify import ClassificationExample
+from cohere.classify import Example  # Fixed import
 
 # Initialize the Cohere client using your provided API key.
 # (For security, consider loading the API key from environment variables.)
@@ -18,10 +18,10 @@ co = cohere.Client(COHERE_API_KEY)
 
 # Define some labeled examples for classification.
 cohere_examples = [
-    ClassificationExample(text="generate a report on unemployment", label="report"),
-    ClassificationExample(text="create a report on market trends", label="report"),
-    ClassificationExample(text="generate diagram of sales data", label="diagram"),
-    ClassificationExample(text="create a diagram of quarterly revenue", label="diagram")
+    Example(text="generate a report on unemployment", label="report"),
+    Example(text="create a report on market trends", label="report"),
+    Example(text="generate diagram of sales data", label="diagram"),
+    Example(text="create a diagram of quarterly revenue", label="diagram")
 ]
 
 def cohere_parse_command(command):