iteration 2

app.py

@@ -1,26 +1,46 @@
 import gradio as gr
 
 # Step 1: Textbox input
-# For now, we wire the textbox to a dummy function. We'll later replace `dummy_inference` with your model call.
+# Define relation types and sample text
+rels = [
+    'acronym',
+    'author',
+    'data description',
+    'data geography',
+    'data source',
+    'data type',
+    'publication year',
+    'publisher',
+    'reference year',
+    'version'
+]
+sample_text = (
+    "Recent studies on ocean currents from the Global Ocean Temperature Dataset (GOTD) "
+    "indicate significant shifts in marine biodiversity."
+)
+
+# Dummy inference echoes input + relations
 def dummy_inference(query: str) -> str:
     # TODO: replace with actual NER+RE model inference
-    return f"Model received: '{query}'"
+    return f"Model received: '{query}' with relations: {rels}"
 
 with gr.Blocks(title="Step 1: Input Box Demo") as demo:
-    gr.Markdown("""
-    ## Step 1: Implement a Text Input
-    Enter any text below. On submit, it will echo back the value.  
-    This is where your NER + relation-extraction model will later consume the query.
-    """
+    gr.Markdown(
+        """
+        ## Step 1: Implement a Text Input
+        Enter any text below (prepopulated with a sample).  
+        This is where your NER + relation-extraction model will later consume the query.
+        """
     )
     query_input = gr.Textbox(
         lines=4,
-        placeholder="Type your text here...",
+        value=sample_text,
         label="Input Text",
+        placeholder="Type your text here...",
     )
     submit_btn = gr.Button("Submit")
     output_box = gr.Textbox(
-        lines=2,
+        lines=3,
         label="Echoed Output",
     )
     submit_btn.click(
@@ -29,5 +49,8 @@ with gr.Blocks(title="Step 1: Input Box Demo") as demo:
         outputs=[output_box],
     )
 
-demo.queue(default_concurrency_limit=5)
-demo.launch(debug=True)
+# Launch the demo
+if __name__ == "__main__":
+    demo.queue(default_concurrency_limit=5)
+    demo.launch(debug=True)
+