Update app.py

app.py

@@ -30,7 +30,7 @@ def get_chemical_info(chemical_name):
 
         final_text = "\n".join(output)
 
-        # Save to a file for download
+        # Save to file
         file_path = f"/tmp/{chemical_name.replace(' ', '_')}_info.txt"
         with open(file_path, "w") as f:
             f.write(final_text)
@@ -41,17 +41,21 @@ def get_chemical_info(chemical_name):
         return f"No information found for '{chemical_name}'. Please try a more precise name.", None
 
 
-# Gradio Interface
-demo = gr.Interface(
-    fn=get_chemical_info,
-    inputs=gr.Textbox(label="Enter Chemical Name"),
-    outputs=[
-        gr.Textbox(label="Compound Information", lines=20),
-        gr.File(label="Download as TXT")
-    ],
-    title="ChemQuery: Learn Molecules Easily",
-    description="Enter a chemical name to get formula, synonyms, atomic structure, and more. Powered by PubChem & PubChemPy. Designed for student use in chemistry learning."
-)
-
-if __name__ == "__main__":
-    demo.launch()
+# Gradio Blocks UI
+with gr.Blocks(title="ChemQuery: Learn Molecules Easily") as demo:
+    gr.Markdown("## ChemQuery: Learn Molecules Easily")
+    gr.Markdown(
+        "Enter a chemical name to get formula, synonyms, atomic structure, and more. "
+        "Powered by PubChem & PubChemPy. Designed for student use in chemistry learning."
+    )
+
+    with gr.Row():
+        chemical_input = gr.Textbox(label="Enter Chemical Name", scale=2, lines=1)
+        submit_btn = gr.Button("Submit", scale=1)
+
+    output_text = gr.Textbox(label="Compound Information", lines=25)
+    download_file = gr.File(label="Download as TXT")
+
+    submit_btn.click(get_chemical_info, inputs=chemical_input, outputs=[output_text, download_file])
+
+demo.launch()