Added fix for changing project and logstream from UI

backend/api/main.py

@@ -116,11 +116,10 @@ async def search(
     protection: bool = Form(False),
     hallucination_detection: bool = Form(False),
     induce_hallucination: bool = Form(False),
-    project_name: str = default_project_name,
-    logstream_name: str = default_logstream_name,
-    dataset_name: str = default_dataset_name,
-):
-
+    project_name: str = Form(...),
+    logstream_name: str = Form(...),
+    dataset_name: str = Form(...),
+) -> JSONResponse:
     response, redacted_response, original_response, context_adherence_score, pii_flag = rag_app.run(
         query,
         protect_enabled=protection,

backend/classes/galileo_platform.py

@@ -1,9 +1,11 @@
-from typing import Optional
-
-from dotenv import load_dotenv
 from pydantic import BaseModel
+from dotenv import load_dotenv
+from typing import Any
+from uuid import UUID4
 
 from galileo import GalileoLogger, GalileoScorers, StageType
+from galileo.log_streams import create_log_stream, get_log_stream
+from galileo.projects import create_project, get_project
 from galileo.protect import invoke_protect
 from galileo.stages import create_protect_stage, get_protect_stage
 from galileo_core.schemas.protect.action import OverrideAction
@@ -25,14 +27,26 @@ class GalileoPlatform:
         self.config = config
         self.protect_stage_id = self.get_protect_stage_id()
 
-    def get_logger(self, project_name: str, logstream_name: str):
+    def get_logger(self, project_name: str, logstream_name: str) -> GalileoLogger | None:
         """Get or create a Galileo Logger."""
-        return GalileoLogger(
-            project=project_name,
-            log_stream=logstream_name,
-        )
+        if not get_project(name=project_name):
+            _ = create_project(name=project_name)
+            print(f"Project {project_name} created")
+
+        if not get_log_stream(name=logstream_name, project_name=project_name):
+            _ = create_log_stream(name=logstream_name, project_name=project_name)
+            print(f"Logstream {logstream_name} created in project {project_name}")
+
+        try:
+            return GalileoLogger(
+                project=project_name,
+                log_stream=logstream_name,
+            )
+        except Exception as e:
+            print(f"Failed to create logger: {e}")
+            return None
 
-    def get_protect_stage_id(self):
+    def get_protect_stage_id(self) -> str | UUID4 | None:
         """Get or create a Galileo Protect stage."""
         try:
             protect_stage = get_protect_stage(
@@ -49,7 +63,7 @@ class GalileoPlatform:
             )
             return protect_stage.id
 
-    def run_protect(self, input: str, output: str, logger: Optional[GalileoLogger] = None) -> dict:
+    def run_protect(self, input: str, output: str, logger: GalileoLogger | None = None) -> dict[Any, Any]:
         """Run Galileo Protect on input and output."""
         response = invoke_protect(
             payload=Payload(input=input, output=output),