Update app.py

app.py

@@ -162,31 +162,33 @@ def run_blast(arguments_string: str) -> str:
     return result_response.text
 
 if __name__ == "__main__":
-    if len(sys.argv) < 3:
+    import os
+    if os.path.basename(sys.argv[0]) == os.path.basename(__file__) and len(sys.argv) > 1:
+        arguments_string = " ".join(sys.argv[1:]) # Reconstruct arguments string from command line
+        results = run_blast(arguments_string)
+
+        if results.startswith("Error:"):
+            print(results, file=sys.stderr) # Print errors to stderr
+            if results == "Error: Invalid arguments. Usage: program database query_sequence":
+                sys.exit(1)
+            elif results == "No hits found.":
+                sys.exit(2)
+            elif results.startswith("Search ") and results.endswith("expired."):
+                sys.exit(3)
+            elif results.startswith("Search ") and results.endswith("failed; please report to blast-help@ncbi.nlm.nih.gov."):
+                sys.exit(4)
+            else: # Catch-all for other errors
+                sys.exit(5)
+        else:
+            print(results)
+            sys.exit(0)
+    elif os.path.basename(sys.argv[0]) == os.path.basename(__file__):
+        print("This script is intended to be used as a tool function.", file=sys.stderr)
+        print("When run directly, it requires command-line arguments:", file=sys.stderr)
         print("Usage: python blast_tool.py program database query_sequence", file=sys.stderr)
         print("       query_sequence should be the sequence itself, not a file.", file=sys.stderr)
         sys.exit(1)
 
-    arguments_string = " ".join(sys.argv[1:]) # Reconstruct arguments string from command line
-    results = run_blast(arguments_string)
-
-    if results.startswith("Error:"):
-        print(results, file=sys.stderr) # Print errors to stderr
-        if results == "Error: Invalid arguments. Usage: program database query_sequence":
-            sys.exit(1)
-        elif results == "No hits found.":
-            sys.exit(2)
-        elif results.startswith("Search ") and results.endswith("expired."):
-            sys.exit(3)
-        elif results.startswith("Search ") and results.endswith("failed; please report to blast-help@ncbi.nlm.nih.gov."):
-            sys.exit(4)
-        else: # Catch-all for other errors
-            sys.exit(5)
-    else:
-        print(results)
-        sys.exit(0)
-
-
 
 final_answer = FinalAnswerTool()