Upload 2 files

app.py

@@ -1,14 +1,28 @@
+import socket
 import gradio as gr
 import pubchempy as pcp
 
+# PubChemPy has no built-in request timeout, so a slow/unresponsive PubChem
+# API call can hang a worker indefinitely. A global socket timeout prevents
+# that from ever blocking the app.
+socket.setdefaulttimeout(15)
+
+
 def get_chemical_info(chemical_name):
+    if not chemical_name or not chemical_name.strip():
+        return "Please enter a chemical name.", None
+
     try:
-        compound = pcp.get_compounds(chemical_name, 'name')[0]
+        compounds = pcp.get_compounds(chemical_name, 'name')
+        if not compounds:
+            return f"No information found for '{chemical_name}'. Please try a more precise name.", None
+
+        compound = compounds[0]
 
         output = []
         output.append(f"IUPAC Name: {compound.iupac_name}")
-        output.append(f"Common Name: {compound.synonyms[0]}")
-        output.append(f"Synonyms: {', '.join(compound.synonyms[:4])}")
+        output.append(f"Common Name: {compound.synonyms[0] if compound.synonyms else 'N/A'}")
+        output.append(f"Synonyms: {', '.join(compound.synonyms[:4]) if compound.synonyms else 'N/A'}")
         output.append(f"Formula: {compound.molecular_formula}")
         output.append(f"Molecular Weight: {compound.molecular_weight}")
         output.append(f"Exact Molecular Weight: {compound.exact_mass}")
@@ -31,7 +45,9 @@ def get_chemical_info(chemical_name):
         final_text = "\n".join(output)
 
         # Save to a file
-        file_path = f"/tmp/{chemical_name.replace(' ', '_')}_info.txt"
+        safe_name = "".join(c for c in chemical_name if c.isalnum() or c in (" ", "_", "-")).strip()
+        safe_name = safe_name.replace(" ", "_") or "compound"
+        file_path = f"/tmp/{safe_name}_info.txt"
         with open(file_path, "w") as f:
             f.write(final_text)
 
@@ -39,26 +55,32 @@ def get_chemical_info(chemical_name):
 
     except IndexError:
         return f"No information found for '{chemical_name}'. Please try a more precise name.", None
+    except (socket.timeout, TimeoutError):
+        return "PubChem took too long to respond. Please try again in a moment.", None
+    except Exception as e:
+        # Catch-all so an unexpected PubChem/network error never crashes the
+        # request handler or shows a raw traceback to the user.
+        return f"Something went wrong while looking up '{chemical_name}': {e}", None
 
 
 # Gradio UI
 with gr.Blocks(title="ChemQuery: Learn Molecules Easily") as demo:
     gr.Markdown("## ChemQuery: Learn Molecules Easily")
     gr.Markdown(
-    """
-    Enter a chemical name to explore its molecular formula, synonyms, atomic structure, and bonding details.
-    ChemQuery is Powered by PubChem & PubChemPy — designed to make chemistry learning simple and fun for students, teachers, and curious minds alike.
-    
-    If you find this initiative helpful, please:
-    - ❤️ Give it a like
-    - 🔁 Share it with your friends, classmates, or educators
-    - ⭐ Star the GitHub repo: [ChemQuery](https://github.com/RezuwanHassan262/ChemQuery/tree/main)
-
-    > _“ChemQuery is a free and open-source chemistry learning tool built as part of an educational initiative to support students worldwide.”_
-    
-    Let's spread the love for science and open knowledge! 🌍🧪✨
-    
-    """
+        """
+        Enter a chemical name to explore its molecular formula, synonyms, atomic structure, and bonding details.
+        ChemQuery is Powered by PubChem & PubChemPy — designed to make chemistry learning simple and fun for students, teachers, and curious minds alike.
+
+        If you find this initiative helpful, please:
+        - ❤️ Give it a like
+        - 🔁 Share it with your friends, classmates, or educators
+        - ⭐ Star the GitHub repo: [ChemQuery](https://github.com/RezuwanHassan262/ChemQuery/tree/main)
+
+        > _“ChemQuery is a free and open-source chemistry learning tool built as part of an educational initiative to support students worldwide.”_
+
+        Let's spread the love for science and open knowledge! 🌍🧪✨
+
+        """
     )
 
     with gr.Row():
@@ -81,5 +103,7 @@ with gr.Blocks(title="ChemQuery: Learn Molecules Easily") as demo:
     download_file = gr.File(label="Download as TXT")
 
     submit_btn.click(get_chemical_info, inputs=chemical_input, outputs=[output_text, download_file])
+    chemical_input.submit(get_chemical_info, inputs=chemical_input, outputs=[output_text, download_file])
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -1,2 +1 @@
 pubchempy==1.0.4
-gradio==4.28.3