Update llmConnect.py

llmConnect.py

@@ -6,6 +6,13 @@ import json
 load_dotenv()
 import csv 
 
+from supabase import create_client, Client
+
+url: str = os.environ.get("SUPABASE_URL")
+key: str = os.environ.get("SUPABASE_KEY")
+supabase: Client = create_client(url, key)
+
+
 
 try:
     client = genai.Client(api_key=os.getenv('geminiAPI'))
@@ -13,6 +20,12 @@ except Exception as e:
     print("Error: ", e)
 
 
+
+
+except Exception as e:
+    print("Error: ", e)
+
+
 def readSystemInstructions():
     try:
         with open('systemInstruction.txt', 'r') as file:
@@ -38,15 +51,44 @@ def generateSyntheticData(topic:str):
 
 
 
-
-def addToDataset(topic:str):
+def addToDataset(topic: str):
     data = generateSyntheticData(topic)
-    with open('dataset.csv', 'w') as file:
-        writer = csv.writer(file)
-        for i in data:
-            for question_number, question_details in i.items():
-                writer.writerow([question_number,question_details['question'],question_details['options'][0],question_details['options'][1],question_details['options'][2],question_details['options'][3],question_details['correct_choice']])
-            print("Added to dataset")
+    if not data:
+        print("Failed to generate data")
+        return
+        
+    try:
+        with open('dataset.csv', 'w', newline='') as file:
+            writer = csv.writer(file)
+            for i in data:
+                for question_number, question_details in i.items():
+                    row = [
+                        question_number,
+                        question_details['question'],
+                        question_details['options'][0],
+                        question_details['options'][1],
+                        question_details['options'][2],
+                        question_details['options'][3],
+                        question_details['correct_choice']
+                    ]
+                    writer.writerow(row)
+                    
+                    try:
+                        supabase.table("quizQuestions").insert({
+                            "topic": topic,
+                            "question": question_details['question'],
+                            "choice-1": question_details['options'][0],
+                            "choice-2": question_details['options'][1],
+                            "choice-3": question_details['options'][2],
+                            "choice-4": question_details['options'][3],
+                            "correctChoice": question_details['correct_choice']
+                        }).execute()
+                    except Exception as e:
+                        print(f"Error inserting to Supabase: {e}")
+                        
+            print("Added to dataset successfully")
+    except Exception as e:
+        print(f"Error writing to CSV: {e}")
 
 
 if __name__ == "__main__":