Update app.py

app.py

@@ -244,7 +244,7 @@ def load_model_and_tokenizers():
             logging.warning(
                 "CUDA is available, but forcing CPU for Gradio app simplicity. Modify if GPU is intended."
             )
-            device = torch.device("cpu")  # Uncomment if GPU is intended
+            device = torch.device("cuda")  # Uncomment if you want to use GPU
             # device = torch.device("cuda")
             # logging.info("CUDA available, using GPU.")
         else:
@@ -521,53 +521,52 @@ Translation uses **greedy decoding** (picks the most likely next word at each st
 """
 
 
-
-# Replace your Interface code with this:
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="cyan")) as iface:
+# Use gr.Blocks exclusively
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="cyan")) as iface: # 'iface' is created here
     gr.Markdown(f"# {title}")
     gr.Markdown(description)
-    
+
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1): # Adjust scale if needed
             smiles_input = gr.Textbox(
                 label="SMILES String",
                 placeholder="Enter SMILES string here (e.g., CCO for Ethanol)",
                 lines=1,
+                # interactive=True # Default is True
             )
-            
-            # Add a button for manual triggering
             submit_btn = gr.Button("Translate")
-            
-        with gr.Column():
+
+        with gr.Column(scale=2): # Give output more space if desired
             output_text = gr.Textbox(
                 label="Predicted IUPAC Name",
-                lines=3,
+                lines=5, # Increased lines slightly
                 show_copy_button=True,
+                # interactive=False # Output shouldn't be user-editable
             )
-    
-    # Connect the events properly
-    submit_btn.click(fn=predict_iupac, inputs=smiles_input, outputs=output_text)
-    
-    # If you still want change event (auto-translation as user types)
-    smiles_input.change(fn=predict_iupac, inputs=smiles_input, outputs=output_text)
-# Output component
-output_text = gr.Textbox(
-    label="Predicted IUPAC Name",
-    lines=3,
-    show_copy_button=True,  # Reduced lines slightly
-)
 
+    # --- Define Event Listeners INSIDE the gr.Blocks context ---
+    # When the button is clicked, call predict_iupac
+    submit_btn.click(
+        fn=predict_iupac,
+        inputs=smiles_input,
+        outputs=output_text,
+        api_name="translate_smiles" # Optional: name for API endpoint
+    )
+
+    # Optional: Trigger prediction when text changes (can be resource-intensive)
+    # If you uncomment this, consider adding a debounce or throttle if using Gradio >= 3.20
+    # smiles_input.change(fn=predict_iupac, inputs=smiles_input, outputs=output_text)
+
+    # Optional: Trigger prediction when text is submitted (e.g., pressing Enter)
+    smiles_input.submit(
+        fn=predict_iupac,
+        inputs=smiles_input,
+        outputs=output_text
+    )
 
-# Create the interface instance
-iface = gr.Interface(
-    fn=predict_iupac,  # The function to call
-    inputs=smiles_input,  # Input component directly
-    outputs=output_text,  # Output component
-    title=title,
-    description=description,
-    allow_flagging="never",
-    theme=gr.themes.Soft(primary_hue="blue", secondary_hue="cyan"),
-)
 # --- Launch the App ---
+# The 'iface' variable is already defined by the 'with gr.Blocks(...)' statement
 if __name__ == "__main__":
-    iface.launch()
+    # You can add server_name="0.0.0.0" if running in Docker/Spaces
+    # and share=True if you want a public link (usually handled by Spaces automatically)
+    iface.launch()