updating test_supabase, updating dockerfile and requirements.txt

Dockerfile

@@ -10,9 +10,8 @@ RUN apt-get update && apt-get install -y git
 # Create a writable cache directory for Hugging Face
 RUN mkdir -p /app/.cache/huggingface/hub && chmod -R 777 /app/.cache
 
-# Install supabase-py directly from GitHub
+# Upgrade pip
 RUN pip install --upgrade pip
-RUN pip install git+https://github.com/supabase/supabase-py.git
 
 # Copy dependencies
 COPY requirements.txt .

requirements.txt

@@ -12,4 +12,5 @@ numpy==1.26.3
 pytest==7.4.3
 pytest-asyncio==0.21.1
 httpx==0.25.2
+supabase==2.13.0
 configparser>=6.0.0

tests/test_supabase.py

@@ -1,5 +1,6 @@
 import configparser
-from supabase import create_client, AsyncClient
+import asyncio
+from supabase import AsyncClient
 
 # Read the properties file
 config = configparser.ConfigParser()
@@ -13,18 +14,65 @@ SUPABASE_KEY = config.get('DEFAULT', 'SUPABASE_KEY')
 supabase = AsyncClient(SUPABASE_URL, SUPABASE_KEY)
 
 # Test data to insert and update
-test_url = "https://www.straitstimes.com/world/united-states/us-senate-confirms-trump-loyalist-kash-patel-to-head-fbi"
-test_headline = "US Senate confirms Trump loyalist Kash Patel to head FBI"
-test_content = "This is some test content for the article."
+test_url = "https://apnews.com/article/trump-air-force-one-boeing-plane-355ed87b00d7d82a061297f68c4ed89b"
+test_headline = "Trump says he’s considering buying used planes to serve as Air Force One amid Boeing delays"
+test_content = "WASHINGTON (AP) — President Donald Trump said Wednesday he is considering buying used Boeing aircraft — perhaps from an overseas seller — to use as Air Force One when he’s aboard, as he fumes over the U.S. plane-maker’s delays in producing two specially modified ones for presidential use."
+
+# Full test data with all fields
+test_data = {
+    "url": test_url,
+    "headline": test_headline,
+    "content": test_content,
+    "sentiment": "Neutral",
+    "bias": "Strongly Left",
+    "bias_score": -1.0,
+    "bias_percentage": 100.0,
+    "flagged_phrases": [],
+    "media_score": {
+        "media_unmasked_score": 49.6,
+        "rating": "Misleading",
+        "details": {
+            "headline_analysis": {
+                "headline_vs_content_score": 38.5,
+                "contradictory_phrases": []
+            },
+            "sentiment_analysis": {
+                "sentiment": "Neutral",
+                "manipulation_score": 0.0,
+                "flagged_phrases": []
+            },
+            "bias_analysis": {
+                "bias": "Strongly Left",
+                "bias_score": -1.0,
+                "bias_percentage": 100.0
+            },
+            "evidence_analysis": {
+                "evidence_based_score": 60.0
+            }
+        }
+    }
+}
 
 # 1. Insert data into the 'article_analysis' table
-def insert_data():
+async def insert_data():
+    try:
+        # Use upsert with conflict on the 'url' field
+        response = await supabase.table('article_analysis').upsert(test_data, on_conflict=['url']).execute()
+        print("Data inserted or updated successfully")
+        print(f"Response data: {response.data}")
+    except Exception as e:
+        print(f"Error inserting or updating data: {str(e)}")
+
+# 2. Update data (e.g., changing the sentiment)
+async def update_data():
     try:
-        response = supabase.table('article_analysis').upsert({
+        updated_sentiment = "Positive"
+        # Ensure that we are not leaving any required fields as null
+        response = await supabase.table('article_analysis').upsert({
             'url': test_url,
+            'sentiment': updated_sentiment,
             'headline': test_headline,
             'content': test_content,
-            'sentiment': 'Neutral',
             'bias': 'Neutral',
             'bias_score': 0.0,
             'bias_percentage': 0.0,
@@ -39,29 +87,16 @@ def insert_data():
                     'evidence_analysis': {'evidence_based_score': 80.0}
                 }
             }
-        }).execute()
-        print("Data inserted successfully")
-        print(f"Response data: {response.data}")
-    except Exception as e:
-        print(f"Error inserting data: {str(e)}")
-
-# 2. Update data (e.g., changing the sentiment)
-def update_data():
-    try:
-        updated_sentiment = "Positive"
-        response = supabase.table('article_analysis').upsert({
-            'url': test_url,
-            'sentiment': updated_sentiment  # Update only the sentiment field
-        }).execute()
+        }, on_conflict=['url']).execute()  # Use on_conflict to handle the conflict by updating the existing URL row
         print("Data updated successfully")
         print(f"Response data: {response.data}")
     except Exception as e:
         print(f"Error updating data: {str(e)}")
 
 # 3. Retrieve data by URL
-def retrieve_data():
+async def retrieve_data():
     try:
-        result = supabase.table('article_analysis').select('*').eq('url', test_url).execute()
+        result = await supabase.table('article_analysis').select('*').eq('url', test_url).execute()
         if result.data:
             print(f"Retrieved data: {result.data}")
         else:
@@ -70,6 +105,11 @@ def retrieve_data():
         print(f"Error retrieving data: {str(e)}")
 
 # Run the tests: Insert, Update, and Retrieve data
-insert_data()
-update_data()
-retrieve_data() 
+async def run_tests():
+    await insert_data()
+    await update_data()
+    await retrieve_data()
+
+# Execute the async function using asyncio
+if __name__ == "__main__":
+    asyncio.run(run_tests())