test: added testcase for test_get_id_from_string

.gitignore

@@ -0,0 +1 @@
+__pycache__

app.py

@@ -130,30 +130,29 @@ EXPLANATION
     
     return out_text
 
-# --- UI ---
-with gr.Blocks(theme='gstaff/xkcd') as demo:
-    gr.Markdown("# xkcd Comic Finder")
-    gr.Markdown(
-        "Sign in with your Hugging Face account so the app can call the model via the Inference API."
-        "\n\n> If you deploy to a Space, add `hf_oauth: true` in your Space metadata and grant the `inference:api` scope."
-    )
-    gr.LoginButton()  # Shows “Sign in with Hugging Face”
-
-    gr.ChatInterface(
-        fn=respond,
-        title="xkcd Comic Finder",
-        description="Find the most suitable xkcd comic for your situation. Use the login button above.",
-        examples=[
-            "I need a comic about procrastination.",
-            "A comic for programmers debugging code.",
-            "Life advice in comic form.",
-        ],
-        type="messages",
-    )
-
 if __name__ == "__main__":
-    global index
-    global meta 
-    index, meta = get_index()
-    embedder = SentenceTransformer("all-MiniLM-L6-v2")
-    demo.launch()
+    # --- UI ---
+    with gr.Blocks(theme='gstaff/xkcd') as demo:
+        gr.Markdown("# xkcd Comic Finder")
+        gr.Markdown(
+            "Sign in with your Hugging Face account so the app can call the model via the Inference API."
+            "\n\n> If you deploy to a Space, add `hf_oauth: true` in your Space metadata and grant the `inference:api` scope."
+        )
+        gr.LoginButton()  # Shows “Sign in with Hugging Face”
+
+        gr.ChatInterface(
+            fn=respond,
+            title="xkcd Comic Finder",
+            description="Find the most suitable xkcd comic for your situation. Use the login button above.",
+            examples=[
+                "I need a comic about procrastination.",
+                "A comic for programmers debugging code.",
+                "Life advice in comic form.",
+            ],
+            type="messages",
+        )
+        global index
+        global meta 
+        index, meta = get_index()
+        embedder = SentenceTransformer("all-MiniLM-L6-v2")
+        demo.launch()

requirements.txt

@@ -3,3 +3,4 @@ faiss-cpu
 transformers
 sentence-transformers
 datasets
+gradio[oauth]

test/test_app.py

@@ -1,12 +0,0 @@
-import pytest
-from app import build_index, respond
-
-def test_build_index():
-    index, meta = build_index()
-    assert index is not None
-    assert meta is not None
-
-def test_respond():
-    respond("I need a comic about procrastination.")
-    assert response is str
-

test_app.py

@@ -0,0 +1,15 @@
+import unittest
+
+from app import get_id_from_string
+
+class TestGetIdFromString(unittest.TestCase):
+    def test_get_id_from_string(self):
+        test_string = """
+    [844] https://xkcd.com/844/
+
+    This comic fits best because it directly addresses the topic of programmers debugging code, specifically the challenges and frustrations that come with it. The comic's flowchart and accompanying text poke fun at the common experience of programmers struggling to write good code, and the inevitable cycle of rewriting and restarting that often ensues.
+    """
+        self.assertEqual(get_id_from_string(test_string), "844")
+
+if __name__ == "__main__":
+    unittest.main()