move logging to its own module

app.py

@@ -1,6 +1,6 @@
 import uuid
+from logging import get_experiment, get_experiment_status, start_experiment
 
-import comet_ml
 import gradio as gr
 import pandas as pd
 from PIL import Image
@@ -16,82 +16,15 @@ You can use this Space to log predictions to [Comet](https://www.comet.ml/site)
 
 Keep track of all your prompts and generated images so that you remember the good ones!
 
-Set your Comet credentials in the Comet Settings tab and create an Experiment for logging data.
-If you want to continue logging to the same Experiment over multiple sessions, add in the
-
-Then use the path to a Space to generate from in the Diffusion Model tab
-"""
+Set your Comet credentials in the Comet Settings tab and create an Experiment for logging data. If you don't have credentials yet,
+you can [sign up for Comet here](https://www.comet.ml/signup)
 
+If you want to continue logging to the same Experiment over multiple sessions, simply provide the experiment name.
 
-def start_experiment(
-    comet_api_key,
-    comet_workspace,
-    comet_project_name,
-    comet_experiment_name,
-    experiment,
-):
-    if not comet_api_key:
-        experiment = None
-        return (
-            experiment,
-            """
-            Please add your API key in order to log your predictions to a Comet Experiment.
-            If you don't have a Comet account yet, you can sign up using the link below:
-
-            https://www.comet.ml/signup
-            """,
-        )
+Set a path to a Space using that uses a Diffusion model and submit your prompt in the Diffusion Model tab
 
-    try:
-        if comet_experiment_name:
-            api_experiment = get_experiment(
-                {
-                    "api_key": comet_api_key,
-                    "workspace": comet_workspace,
-                    "project_name": comet_project_name,
-                    "experiment": comet_experiment_name,
-                }
-            )
-        else:
-            api_experiment = comet_ml.APIExperiment(
-                api_key=comet_api_key,
-                workspace=comet_workspace,
-                project_name=comet_project_name,
-            )
-        experiment = {
-            "api_key": comet_api_key,
-            "workspace": comet_workspace,
-            "project_name": comet_project_name,
-            "experiment": api_experiment.name,
-        }
-
-        return experiment, f"Started {api_experiment.name}. Happy logging!😊"
-
-    except Exception as e:
-        return None, e
-
-
-def get_experiment(experiment_state):
-    try:
-        api_key = experiment_state.get("api_key")
-        workspace = experiment_state.get("workspace")
-        project = experiment_state.get("project_name")
-        experiment_name = experiment_state.get("experiment")
-
-        return comet_ml.API(api_key=api_key).get_experiment(
-            workspace=workspace, project_name=project, experiment=experiment_name
-        )
-    except Exception as e:
-        return None
-
-
-def get_experiment_status(experiment_state):
-    experiment = get_experiment(experiment_state)
-    if experiment is not None:
-        name = experiment.name
-        return experiment_state, f"Currently logging to: {name}"
-
-    return experiment_state, f"No Experiments found"
+** Note: ** This Space will still run even if you don't set credentials
+"""
 
 
 def predict(
@@ -184,8 +117,8 @@ def start_interface():
                     only text as input and produces an image as an output
                     """
                 )
-                model = gr.Textbox(label="Model", value="spaces/valhalla/glide-text2im")
-                prompt = gr.Textbox(label="Prompt")
+                model = gr.Textbox(label="Model", value="spaces/valhalla/glide-text2im", placeholder="Enter a path to a Space")
+                prompt = gr.Textbox(label="Prompt", value="an oil painting of a corgi", placeholder="Enter your text prompt here")
 
                 outputs = gr.Image(label="Image")
 

logging.py

@@ -0,0 +1,74 @@
+import comet_ml
+
+
+def start_experiment(
+    comet_api_key,
+    comet_workspace,
+    comet_project_name,
+    comet_experiment_name,
+    experiment,
+):
+    if comet_api_key is None:
+        experiment = None
+        return (
+            experiment,
+            """
+            Please add your API key in order to log your predictions to a Comet Experiment.
+            If you don't have a Comet account yet, you can sign up using the link below:
+
+            https://www.comet.ml/signup
+            """,
+        )
+
+    try:
+        if comet_experiment_name:
+            # Retrieve the Experiment if it already exists
+            api_experiment = get_experiment(
+                {
+                    "api_key": comet_api_key,
+                    "workspace": comet_workspace,
+                    "project_name": comet_project_name,
+                    "experiment": comet_experiment_name,
+                }
+            )
+        else:
+            # Create a new Experiment
+            api_experiment = comet_ml.APIExperiment(
+                api_key=comet_api_key,
+                workspace=comet_workspace,
+                project_name=comet_project_name,
+            )
+        experiment = {
+            "api_key": comet_api_key,
+            "workspace": comet_workspace,
+            "project_name": comet_project_name,
+            "experiment": api_experiment.name,
+        }
+
+        return experiment, f"Started {api_experiment.name}. Happy logging!😊"
+
+    except Exception as e:
+        return None, e
+
+
+def get_experiment(experiment_state):
+    try:
+        api_key = experiment_state.get("api_key")
+        workspace = experiment_state.get("workspace")
+        project = experiment_state.get("project_name")
+        experiment_name = experiment_state.get("experiment")
+
+        return comet_ml.API(api_key=api_key).get_experiment(
+            workspace=workspace, project_name=project, experiment=experiment_name
+        )
+    except Exception as e:
+        return None
+
+
+def get_experiment_status(experiment_state):
+    experiment = get_experiment(experiment_state)
+    if experiment is not None:
+        name = experiment.name
+        return experiment_state, f"Currently logging to: {name}"
+
+    return experiment_state, f"No Experiments found"