Update app.py

app.py

@@ -28,16 +28,27 @@ def get_chemical_info(chemical_name):
             atom2 = atom_dict.get(bond.aid2, f"Atom {bond.aid2}")
             output.append(f"  Bond between {atom1} and {atom2}, Order: {bond.order}")
 
-        return "\n".join(output)
+        final_text = "\n".join(output)
+
+        # Save to a file for download
+        file_path = f"/tmp/{chemical_name.replace(' ', '_')}_info.txt"
+        with open(file_path, "w") as f:
+            f.write(final_text)
+
+        return final_text, file_path
 
     except IndexError:
-        return f"No information found for '{chemical_name}'. Please try a more precise 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),
+    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."
 )