feat: support timer and storage

app.py

@@ -6,11 +6,104 @@ import openai
 import logging
 from gtts import gTTS
 import re
+from shillelagh.backends.apsw.db import connect
+
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Google Sheets API setup
+credentials_json = os.getenv('GOOGLE_API_JSON_CONTENT')
+# Load the credentials from the JSON string
+credentials_info = json.loads(credentials_json)
+# credentials = service_account.Credentials.from_service_account_info(credentials_info)
+
+# os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = 'location-455719-d08225f31b58.json'
+
+# URL format: https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit
+#  https://docs.google.com/spreadsheets/d/1CMdSSTxjw9ooYDzmAufL0j3EtUxAQ6vdwbAL5e3t7aY/edit?usp=sharing
+spreadsheet_id = '1CMdSSTxjw9ooYDzmAufL0j3EtUxAQ6vdwbAL5e3t7aY'
+sheet_table = {
+    "Location": 0,
+    "Config": 1310222727
+}
+location_table = f'https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit?gid={sheet_table["Location"]}'
+config_table = f'https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit?gid={sheet_table["Config"]}'
+
+# Connect to the Google Sheet
+connection = connect(
+    ":memory:", 
+    adapter_kwargs={
+        "gsheetsapi": {
+            # "service_account_file": 'location-455719-d08225f31b58.json'
+            "service_account_info": credentials_info,
+        }
+    }
+)
+cursor = connection.cursor()
+
+# Get latest config
+rows = cursor.execute(f"SELECT * FROM '{config_table}' LIMIT 1").fetchall()
+config = rows[0][1]
+config = json.loads(config)
+config["last_run"] = config.get("last_run", 0) + 1
+updated_config = json.dumps(config)
+try:
+    cursor.execute(
+        f"UPDATE '{config_table}' SET config = ? WHERE id = ?",
+        (updated_config, 1)
+    )
+except Exception as e:
+    logger.error(f"Error updating config: {e}")
+sample = 0
+def update_location(obj):
+    global cursor
+    global config
+    global location_table
+    global sample
+    sample += 1
+    try:
+        # Update the Google Sheet with the new location
+        cursor.execute(
+            f"INSERT INTO '{location_table}' (Run, Sample, LocationObj, Address, AI) VALUES (?, ?, ?, ?, ?)", 
+            (config["last_run"], sample, json.dumps(obj), obj["address"], obj["ai_info"])
+        )
+        connection.commit()
+        logger.info(f"Updated location in Google Sheets: {obj}")
+    except Exception as e:
+        logger.error(f"Error updating location in Google Sheets: {e}")
+
+# Global state for timer control
+is_timer_active = False
+current_interval = 10.0  # Default to 1 second
+timer = None
+
+
+
+def toggle_timer(state):
+    global is_timer_active, timer
+    is_timer_active = state
+    if state:
+        # Start the timer
+        timer = gr.Timer(current_interval)
+    # return f"Timer {'activated' if state else 'deactivated'}."
+
+
+counter = 0
+def auto_trigger():
+    """
+    Simulates pressing the '📍 Get My Location and Street Info' button.
+    This function is triggered by the timer.
+    """
+    global counter
+    global is_timer_active
+    counter += 1
+    if is_timer_active:
+        return f"Auto-triggered: Getting location...{counter}"
+    return None
+
+
 def text_to_speech(text, lang='en'):
     """
     Converts text to speech using gTTS and returns the path to the audio file.
@@ -132,16 +225,21 @@ def get_ai_info(street_name):
         {"role": "user", "content": f"Location information:{street_name}."}
     ]
     try:
+        # TODO: remove commentted code 
         client = openai.OpenAI()
         completion = client.chat.completions.create(
-            model="gpt-4o-search-preview",
+            model="gpt-4o-mini-search-preview",
             web_search_options={
                 "search_context_size": "low",
             },
             messages=messages,
             # max_tokens=150,
         )
-        return completion.choices[0].message.content.strip()
+        j = completion
+        return j
+        r = completion.choices[0].message.content.strip()
+        return r
+        # return "Debug text"
     except Exception as e:
         return f"Error retrieving street information: {e}"
 
@@ -152,11 +250,15 @@ def preprocess_text(input_text):
     return text_without_urls
 
 
+
 def process_location(coords):
     """
     Processes the user's coordinates to retrieve the address, street information,
     and nearby points of interest.
     """
+    global is_timer_active
+    if not is_timer_active:
+        return "Timer is not active. Please enable the timer to get location updates.", "Unable to retrieve street information.", None
     try:
         lat, lon = map(float, coords.split(","))
         address_data = get_address(lat, lon)
@@ -169,14 +271,27 @@ def process_location(coords):
         poi_descriptions = '; '.join(f"{poi['name']} ({poi['type']})" for poi in pois)
         combined_info = f"{address} Nearby points of interest include: {poi_descriptions}."
         logging.info(f"Retrieved address: {combined_info}")
-        ai_info = get_ai_info(address)
+        ai_obj = get_ai_info(address)
+        ai_info = ai_obj.choices[0].message.content.strip()
+        ai_obj = ai_obj.to_json()
         ai_info = preprocess_text(ai_info)
-        audio_file = None #text_to_speech(ai_info)
+        audio_file = None #TODO text_to_speech(ai_info)
+        location_obj = {
+            "lat": lat,
+            "lon": lon,
+            "address": address,
+            "combined_info": combined_info,
+            "ai_obj": ai_obj,
+            "ai_info": ai_info,
+        }
+        # Update the Google Sheet with the new location
+        update_location(location_obj)
         return combined_info, ai_info, audio_file
     except ValueError:
         return "Invalid coordinates provided.", "Unable to retrieve street information.", None
 
 
+
 # Gradio Interface
 with gr.Blocks() as demo:
     coords_box = gr.Textbox(label="Your Coordinates (Lat, Lon)", visible=False)
@@ -187,9 +302,42 @@ with gr.Blocks() as demo:
     with gr.Row():
         get_location_btn = gr.Button("📍 Get My Location and Street Info")
         read_aloud_btn = gr.Button("🔊 Read Aloud")
+        toggle_button = gr.Checkbox(label="Enable Auto Trigger", value=False)
+        interval_selector = gr.Dropdown(
+            choices=["10 sec", "1 min", "2 min", "5 min"], value="10 sec", label="Set Timer Interval"
+        )
+
 
     # JavaScript to fetch geolocation and populate coords_box
-    get_location_btn.click(None, js="""
+
+    debug_geolocation_js = """
+    () => new Promise((resolve, reject) => {
+        try {
+            // Generate realistic coordinates within populated areas
+            const locations = [
+                {lat: 40.7128, lon: -74.0060}, // New York, USA
+                {lat: 48.8566, lon: 2.3522},   // Paris, France
+                {lat: 35.6762, lon: 139.6503}, // Tokyo, Japan
+                {lat: 51.5074, lon: -0.1278},  // London, UK
+                {lat: -33.8688, lon: 151.2093},// Sydney, Australia
+                {lat: 19.0760, lon: 72.8777},  // Mumbai, India
+                {lat: -23.5505, lon: -46.6333} // São Paulo, Brazil
+            ];
+            
+            // Select a random location
+            const randomLocation = locations[Math.floor(Math.random() * locations.length)];
+            const coords = `${randomLocation.lat.toFixed(6)},${randomLocation.lon.toFixed(6)}`;
+            
+            console.log("Debug Geolocation fetched:", coords);
+            resolve(coords);
+        } catch (error) {
+            console.error("Error generating random geolocation: " + error.message);
+            reject();
+        }
+    })
+    """
+
+    real_geolocation_js = """
         () => new Promise((resolve, reject) => {
             if (navigator.geolocation) {
                 navigator.geolocation.getCurrentPosition(
@@ -208,7 +356,11 @@ with gr.Blocks() as demo:
                 reject();
             }
         })
-    """, outputs=coords_box)
+    """
+
+    geolocation_js = real_geolocation_js
+    
+    get_location_btn.click(None, js=geolocation_js, outputs=coords_box)
 
     # Update outputs when coordinates change
     coords_box.change(
@@ -224,5 +376,16 @@ with gr.Blocks() as demo:
         outputs=audio_output
     )
 
+    # Toggle Timer On/Off
+    toggle_button.change(toggle_timer, inputs=toggle_button, outputs=None)
+
+    # Hidden textbox that gets updated by the Timer to trigger JS code
+    trigger_box = gr.Textbox(label="Trigger Box", visible=True)
+    # Timer component - triggers every `current_interval` seconds
+    timer = gr.Timer(config.get("interval", 300))
+    timer.tick(fn=auto_trigger, outputs=trigger_box)
+    trigger_box.change(fn=None, inputs = None, outputs = coords_box, js=geolocation_js)
+
+
 # Launch the Gradio application
 demo.launch()

sheet.py

@@ -0,0 +1,56 @@
+from shillelagh.backends.apsw.db import connect
+from google.oauth2 import service_account
+import os
+import json
+# Set up environment variable for authentication
+# If using a service account, set the environment variable for authentication
+
+credentials_json = os.getenv('GOOGLE_API_JSON_CONTENT')
+# Load the credentials from the JSON string
+credentials_info = json.loads(credentials_json)
+# credentials = service_account.Credentials.from_service_account_info(credentials_info)
+
+# os.environ['GOOGLE_APPLICATION_CREDENTIALS'] = 'location-455719-d08225f31b58.json'
+
+# URL format: https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit
+#  https://docs.google.com/spreadsheets/d/1CMdSSTxjw9ooYDzmAufL0j3EtUxAQ6vdwbAL5e3t7aY/edit?usp=sharing
+spreadsheet_id = '1CMdSSTxjw9ooYDzmAufL0j3EtUxAQ6vdwbAL5e3t7aY'
+sheet_name = 'Config'
+sheet_table = {
+    "Location": 0,
+    "Config": 1310222727
+}
+url = f"https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit?gid={sheet_table[sheet_name]}"
+
+
+# Connect to the Google Sheet
+connection = connect(
+    ":memory:", 
+    adapter_kwargs={
+        "gsheetsapi": {
+            # "service_account_file": 'location-455719-d08225f31b58.json'
+            "service_account_info": credentials_info,
+        }
+    }
+)
+cursor = connection.cursor()
+
+
+# Example query to read data
+rows = cursor.execute(f"SELECT * FROM '{url}' LIMIT 1").fetchall()
+# rowid = rows[0][0]
+config = rows[0][1]
+config = json.loads(config)
+config["last_run"] = config.get("last_run", 0) + 1
+updated_config = json.dumps(config)
+print(rows)
+# cursor.execute(f"DELETE FROM '{url}' ")
+try:
+    cursor.execute(
+        f"UPDATE '{url}' SET config = ? WHERE id = ?",
+        (updated_config, 1)
+    )
+
+except Exception as e:
+    print(f"Error inserting data: {e}")
+