Upload folder using huggingface_hub

ai_responses_demo.py

@@ -74,21 +74,24 @@ ORACLE_API_KEY = os.getenv("ORACLE_KEY", ORACLE_API_KEY)
 TOGETHER_API_KEY = os.getenv("TOGETHER_KEY", TOGETHER_API_KEY)
 OPENAI_API_KEY = os.getenv("OPENAI_KEY", OPENAI_API_KEY)
 
+import pandas as pd
+
 def random_test_case():
     try:
         df = pd.read_csv("test cases.csv")
-        
-        # Check for required columns
-        if not {"user input", "response"}.issubset(df.columns):
-            raise ValueError("CSV must contain 'user input' and 'response' columns.")
-        
-        # Sample one row
+
+        # Print out the first few rows for debugging
+        print("CSV Loaded. First few rows:")
+        print(df.head())
+
+        if "user input" not in df.columns or "response" not in df.columns:
+            raise ValueError(f"CSV must have 'user input' and 'response' columns, found: {df.columns.tolist()}")
+
         sample = df.sample(1).iloc[0]
-        
-        # Return both fields
         return sample["user input"], sample["response"]
-    
+
     except Exception as e:
+        print(f"[ERROR] while loading CSV: {e}")
         return f"Error: {e}", ""
 
 # Custom CSS for styling