Dev & Prod

.github/workflows/main.yml

@@ -10,17 +10,58 @@ env:
   api_secret: ${{ secrets.HF }}
 
 jobs:
-  sync-to-hub:
+  Build:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
+      
+      # Set up Python 3.11 environment
+      - name: Set up Python 3.11
+        uses: actions/setup-python@v4
         with:
-          fetch-depth: 0
+          python-version: "3.11"
+
+      # Install dependencies
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install -e ".[dev]"
+
+      - name: Model Testing
+        run: |
+          python3 inference.py
+  
+  DeployDev:
+    name: Deploy to Dev 
+    if: github.event_name == 'pull_request'
+    needs: [Build]
+    runs-on: ubuntu-latest
+    environment: 
+      name: Dev
+      url: 'https://huggingface.co/spaces/cccmatthew/gp30-dev'
+    steps:
+      - name: Add remote 
+        env:
+          HF: ${{ secrets.HF }}
+        run: git remote add space https://cccmatthew:$HF@huggingface.co/spaces/cccmatthew/gp30-dev
+      - name: Deploy
+        env:
+          HF: ${{ secrets.HF }}
+        run: git push --force https://cccmatthew:$HF@huggingface.co/spaces/cccmatthew/gp30-dev main
+
+  DeployProd:
+    name: Deploy to Production 
+    needs: [DeployStaging]
+    runs-on: ubuntu-latest
+    environment: 
+      name: Prod
+      url: 'https://huggingface.co/spaces/cccmatthew/gp30-deployment'
+    steps:
       - name: Add remote 
         env:
           HF: ${{ secrets.HF }}
         run: git remote add space https://cccmatthew:$HF@huggingface.co/spaces/cccmatthew/gp30-deployment
-      - name: Push to hub
+      - name: Deploy
         env:
           HF: ${{ secrets.HF }}
         run: git push --force https://cccmatthew:$HF@huggingface.co/spaces/cccmatthew/gp30-deployment main

inference.py

@@ -0,0 +1,59 @@
+import streamlit as st
+import requests
+import csv
+import datetime
+import time
+import pandas as pd
+import matplotlib.pyplot as plt
+import os
+
+##########################
+SECRET = os.environ["api_secret"]
+headers = {"Authorization": "Bearer " + SECRET}
+API_URL = "https://api-inference.huggingface.co/models/cccmatthew/surrey-gp30"
+##########################
+
+def log_to_csv(sentence, results, response_time):
+    with open('model_interactions.csv', 'a', newline='') as file:
+        writer = csv.writer(file)
+        writer.writerow([datetime.datetime.now(), sentence, results, response_time])
+
+def send_request_with_retry(url, headers, json_data, retries=3, backoff_factor=1):
+    """Send request with retries on timeouts and HTTP 503 errors."""
+    for attempt in range(retries):
+        start_time = time.time()
+        try:
+            response = requests.post(url, headers=headers, json=json_data)
+            response.raise_for_status()
+            response_time = time.time() - start_time
+            return response, response_time
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 503:
+                st.info('Server is unavailable, retrying...')
+            else:
+                raise
+        except (requests.exceptions.ConnectionError, requests.exceptions.Timeout) as e:
+            st.info(f"Network issue ({str(e)}), retrying...")
+        time.sleep(backoff_factor * (2 ** attempt))
+
+    st.error("Failed to process request after several attempts.")
+    return None, None
+
+sentence = [
+    "RAFs are plotted for a selection of neurons in the dorsal zone (DZ) of auditory cortex in Figure 1.",
+    "Light dissolved inorganic carbon (DIC) resulting from the oxidation of hydrocarbons.",
+    "Images were acquired using a GE 3.0T MRI scanner with an upgrade for echo-planar imaging (EPI)."
+]
+
+API_URL = API_URL
+headers = headers
+response, response_time = send_request_with_retry(API_URL, headers, {"inputs": sentence})
+
+if response is not None:
+    results = response.json()
+    print(results)
+    st.write('Results:')
+    log_to_csv(sentence, results, response_time)
+
+else:
+    st.error("Unable to classify the sentence due to server issues.")